cheanup

1 files changed, 3 insertions, 1 deletions

diff --git a/script/vm/compiler.lua b/script/vm/compiler.lua
index 531a0cd2..1d79cc1b 100644
--- a/script/vm/compiler.lua
+++ b/script/vm/compiler.lua
@@ -60,6 +60,7 @@ local searchFieldSwitch = util.switch()
         end
     end)
     : case 'local'
+    : case 'self'
     : call(function (suri, node, key, pushResult)
         local fields
         if key then
@@ -716,7 +717,7 @@ local function compileLocalBase(source)
         hasMarkDoc = bindDocs(source)
     end
     local hasMarkParam
-    if source.dummy and not hasMarkDoc then
+    if source.type == 'self' and not hasMarkDoc then
         hasMarkParam = true
         vm.setNode(source, vm.compileNode(source.method.node))
     end
@@ -834,6 +835,7 @@ local compilerSwitch = util.switch()
         end
     end)
     : case 'local'
+    : case 'self'
     : call(function (source)
         local baseNode = compileLocalBase(source)
         vm.setNode(source, baseNode, true)