diff --git a/spec/model/model_helper_spec.rb b/spec/model/model_helper_spec.rb
index 913c3183b..2556ac983 100644
--- a/spec/model/model_helper_spec.rb
+++ b/spec/model/model_helper_spec.rb
@@ -10,7 +10,7 @@
     # Oxidized.setup_logger
     @node = MockSsh.get_node('garderos')
     @test = ATOMS::TestOutput.new('garderos', 'R7709_003_006_068')
-    @mockmodel = MockSsh.new(@test)
+    @mockmodel = MockSsh.new(@test.simulation)
     Net::SSH.stubs(:start).returns @mockmodel
   end