diff --git a/spec/workflow_spec.rb b/spec/workflow_spec.rb
index 52dc1085..8c10bf06 100644
--- a/spec/workflow_spec.rb
+++ b/spec/workflow_spec.rb
@@ -62,6 +62,10 @@
 
       expect { described_class.new(payload) }.to raise_error(Floe::InvalidWorkflowError, /must be less than or equal to 80 characters/)
     end
+
+    it "raises an exception for invalid context" do
+      expect { described_class.new(make_payload({"Start" => {"Type" => "Success"}}), "abc") }.to raise_error(Floe::InvalidWorkflowError, /unexpected token/)
+    end
   end
 
   describe "#run_nonblock" do