diff --git a/metals/src/main/scala/scala/meta/internal/metals/ReferenceProvider.scala b/metals/src/main/scala/scala/meta/internal/metals/ReferenceProvider.scala
index 7921ce10069..cf16b0a755a 100644
--- a/metals/src/main/scala/scala/meta/internal/metals/ReferenceProvider.scala
+++ b/metals/src/main/scala/scala/meta/internal/metals/ReferenceProvider.scala
@@ -503,19 +503,19 @@ final class ReferenceProvider(
           )
         else Seq.empty
 
-    val sourceContainsDefinition =
-      occ.role.isDefinition || snapshot.symbols.exists(
-        _.symbol == occ.symbol
-      )
-    val workspaceRefs =
-      workspaceReferences(
-        source,
-        isSymbol,
-        isIncludeDeclaration,
-        findRealRange,
-        includeSynthetics,
-        sourceContainsDefinition,
-      )
+      val sourceContainsDefinition =
+        occ.role.isDefinition || snapshot.symbols.exists(
+          _.symbol == occ.symbol
+        )
+      val workspaceRefs =
+        workspaceReferences(
+          source,
+          isSymbol,
+          isIncludeDeclaration,
+          findRealRange,
+          includeSynthetics,
+          sourceContainsDefinition,
+        )
       Future.successful(local ++ workspaceRefs)
     }
   }
diff --git a/mtags/src/main/scala-2/scala/meta/internal/pc/PcReferencesProvider.scala b/mtags/src/main/scala-2/scala/meta/internal/pc/PcReferencesProvider.scala
index 5f27237db83..522dfa43609 100644
--- a/mtags/src/main/scala-2/scala/meta/internal/pc/PcReferencesProvider.scala
+++ b/mtags/src/main/scala-2/scala/meta/internal/pc/PcReferencesProvider.scala
@@ -59,7 +59,7 @@ class PcReferencesProvider(
                     )
                   )
                 }
-            compiler.unitOfFile.remove(unit.source.file)
+            compiler.unitOfFile.remove(collector.unit.source.file)
             result
           }