Test fix.

2 files changed, 2 insertions, 0 deletions

diff --git a/nomsu.lua b/nomsu.lua
index 0dd9d5f..496cd28 100644
--- a/nomsu.lua
+++ b/nomsu.lua
@@ -339,6 +339,7 @@ do
       if max_operations then
         local timeout
         timeout = function()
+          debug.sethook()
           return self:error("Execution quota exceeded. Your code took too long.")
         end
         debug.sethook(timeout, "", max_operations)
diff --git a/nomsu.moon b/nomsu.moon
index 4728c91..bf5c979 100755
--- a/nomsu.moon
+++ b/nomsu.moon
@@ -277,6 +277,7 @@ class NomsuCompiler
     run: (src, filename, max_operations=nil)=>
         if max_operations
             timeout = ->
+                debug.sethook!
                 @error "Execution quota exceeded. Your code took too long."
             debug.sethook timeout, "", max_operations
         tree = @parse(src, filename)