From 2f1c0284fe81446a603ad0e00fcf5137596156cf Mon Sep 17 00:00:00 2001
From: Bruce Hill <bitbucket@bruce-hill.com>
Date: Mon, 9 Oct 2017 04:37:16 -0700
Subject: [PATCH] Test fix.

---
 nomsu.lua  | 1 +
 nomsu.moon | 1 +
 2 files changed, 2 insertions(+)

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)