2 files changed, 18 insertions, 18 deletions
diff --git a/test/references/all.lua b/test/references/all.lua
index a9442ae1..f8fe3d6e 100644
--- a/test/references/all.lua
+++ b/test/references/all.lua
@@ -211,3 +211,21 @@ function <!A!>:f() end
 
 local <!b!> = <!A!>:f()
 ]]
+
+TEST [[
+local function f()
+    return <~<!function~> ()
+    end!>
+end
+
+local <!f2!> = f()
+]]
+
+TEST [[
+local function f()
+    return nil, <~<!function~> ()
+    end!>
+end
+
+local _, <!f2!> = f()
+]]
diff --git a/test/references/init.lua b/test/references/init.lua
index e90cb2a8..237e41da 100644
--- a/test/references/init.lua
+++ b/test/references/init.lua
@@ -138,24 +138,6 @@ end
 ]]
 
 TEST [[
-local function f()
-    return <~<!function~> ()
-    end!>
-end
-
-local <!f2!> = f()
-]]
-
-TEST [[
-local function f()
-    return nil, <~<!function~> ()
-    end!>
-end
-
-local _, <!f2!> = f()
-]]
-
-TEST [[
 local <?x?>
 local function f()
     return <!x!>