2 files changed, 18 insertions, 2 deletions
diff --git a/test/basic/linker.lua b/test/basic/linker.lua
index bf52c949..3b18ddc9 100644
--- a/test/basic/linker.lua
+++ b/test/basic/linker.lua
@@ -11,8 +11,14 @@ local function getSource(pos)
         or source.type == 'setlocal'
         or source.type == 'setglobal'
         or source.type == 'getglobal'
-        or source.type == 'field'
-        or source.type == 'method'
+        or source.type == 'setfield'
+        or source.type == 'getfield'
+        or source.type == 'setmethod'
+        or source.type == 'getmethod'
+        or source.type == 'tablefield'
+        or source.type == 'setindex'
+        or source.type == 'getindex'
+        or source.type == 'tableindex'
         or source.type == 'label'
         or source.type == 'goto' then
             return source
diff --git a/test/references/init.lua b/test/references/init.lua
index c4e5018a..e24694d2 100644
--- a/test/references/init.lua
+++ b/test/references/init.lua
@@ -96,6 +96,16 @@ local <?a?> = 1
 ]]
 
 TEST [[
+local <!a!>
+local <?b?> = <!a!>
+]]
+
+TEST [[
+local <?a?>
+local <!b!> = <!a!>
+]]
+
+TEST [[
 local t = {
     <!a!> = 1
 }