diff --git a/packages/core/src/codewhisperer/util/supplementalContext/utgUtils.ts b/packages/core/src/codewhisperer/util/supplementalContext/utgUtils.ts
index 19a21acee58..63c29dc1c9a 100644
--- a/packages/core/src/codewhisperer/util/supplementalContext/utgUtils.ts
+++ b/packages/core/src/codewhisperer/util/supplementalContext/utgUtils.ts
@@ -129,7 +129,7 @@ async function findSourceFileByContent(
 
     throwIfCancelled(cancellationToken)
 
-    testElementList.push(...extractClasses(editor.document.fileName, languageConfig.classExtractionPattern))
+    testElementList.push(...extractClasses(testFileContent, languageConfig.classExtractionPattern))
 
     throwIfCancelled(cancellationToken)