Fixed some bugs.

1 files changed, 3 insertions, 0 deletions

diff --git a/nomsu.moon b/nomsu.moon
index 48e8e00..8eb06d8 100755
--- a/nomsu.moon
+++ b/nomsu.moon
@@ -283,6 +283,9 @@ class NomsuCompiler
         @environment.Source = Source
         @environment.ARG_ORDERS = setmetatable({}, {__mode:"k"})
         @environment.ALIASES = setmetatable({}, {__mode:"k"})
+        @environment.compile_time = (fn)->
+            @environment.COMPILE_TIME[fn] = true
+            return fn
         @environment.COMPILE_TIME = {}
         @environment.LOADED = {}
         @environment.AST = AST