diff --git a/nomsu_compiler.lua b/nomsu_compiler.lua
index b5619d2..af0636b 100644
--- a/nomsu_compiler.lua
+++ b/nomsu_compiler.lua
@@ -255,7 +255,7 @@ compile = function(self, tree)
       end
       local line_lua = self:compile(line)
       lua:add(line_lua)
-      if not (line_lua:last(1) == ";" or line_lua:last(4):match("[^a-zA-Z0-9]end$")) then
+      if not (line_lua:last(1) == ";" or line_lua:last(4):match("[^_a-zA-Z0-9]end$")) then
         lua:add(";")
       end
     end
diff --git a/nomsu_compiler.moon b/nomsu_compiler.moon
index 330378f..6fa9ffd 100644
--- a/nomsu_compiler.moon
+++ b/nomsu_compiler.moon
@@ -208,7 +208,7 @@ compile = (tree)=>
                 if i > 1 then lua\add "\n"
                 line_lua = @compile(line)
                 lua\add line_lua
-                unless line_lua\last(1) == ";" or line_lua\last(4)\match("[^a-zA-Z0-9]end$")
+                unless line_lua\last(1) == ";" or line_lua\last(4)\match("[^_a-zA-Z0-9]end$")
                     lua\add ";"
             return lua