diff --git a/lib/output/project_files_creator.rb b/lib/output/project_files_creator.rb
index c1e6f3a38..273deb0cb 100644
--- a/lib/output/project_files_creator.rb
+++ b/lib/output/project_files_creator.rb
@@ -13,7 +13,7 @@ class ProjectFilesCreator
 # @param [Object] systems list of systems that have been defined and randomised
 # @param [Object] out_dir the directory that the project output should be stored into
 # @param [Object] scenario the file path used to as a basis
-  def initialize(systems, out_dir, scenario)
+  def initialize(systems, out_dir, scenario, gui_output)
     @systems = systems
     @out_dir = out_dir
 
@@ -24,6 +24,7 @@ class ProjectFilesCreator
     end
     @scenario = scenario
     @time = Time.new.to_s
+    @gui_output = gui_output
 
   end
 
@@ -86,4 +87,4 @@ class ProjectFilesCreator
     binding
   end
 
-end
\ No newline at end of file
+end