diff --git a/prism/src/prism/TestModelGenerator.java b/prism/src/prism/TestModelGenerator.java
index 8721986e..692909b6 100644
--- a/prism/src/prism/TestModelGenerator.java
+++ b/prism/src/prism/TestModelGenerator.java
@@ -187,6 +187,8 @@ public class TestModelGenerator extends DefaultModelGenerator
 	{
 		try {
 			Prism prism = new Prism(new PrismPrintStreamLog(System.out));
+			prism.setMainLog(new PrismFileLog("stdout"));
+			prism.initialise();
 
 			int test = 2;
 
@@ -212,10 +214,12 @@ public class TestModelGenerator extends DefaultModelGenerator
 				System.out.println(res);
 			}
 
+			prism.closeDown(true);
 		} catch (PrismException e) {
 			System.err.println("Error: " + e.getMessage());
 		} catch (FileNotFoundException e) {
 			System.err.println("Error: " + e.getMessage());
 		}
+
 	}
 }