diff --git a/test/codegen/fromcto/graphql/graphqlvisitor.js b/test/codegen/fromcto/graphql/graphqlvisitor.js
index e608f1c6..a944000a 100644
--- a/test/codegen/fromcto/graphql/graphqlvisitor.js
+++ b/test/codegen/fromcto/graphql/graphqlvisitor.js
@@ -369,6 +369,8 @@ describe('GraphQLVisitor', function () {
             param.fileWriter.writeLine.withArgs(1, 'key: String').calledOnce.should.be.ok;
             param.fileWriter.writeLine.withArgs(1, 'value: String').calledOnce.should.be.ok;
             param.fileWriter.writeLine.withArgs(0, '}').calledOnce.should.be.ok;
+
+            sandbox.restore();
         });
     });