diff --git a/platform/testFramework/src/com/intellij/testFramework/propertyBased/InvokeCompletion.java b/platform/testFramework/src/com/intellij/testFramework/propertyBased/InvokeCompletion.java
index 61619d27c06a..18d00bb77a92 100644
--- a/platform/testFramework/src/com/intellij/testFramework/propertyBased/InvokeCompletion.java
+++ b/platform/testFramework/src/com/intellij/testFramework/propertyBased/InvokeCompletion.java
@@ -107,10 +107,9 @@ public class InvokeCompletion extends ActionOnFile {
                                  char completionChar,
                                  Environment env) {
     int caretOffset = editor.getCaretModel().getOffset();
-    int adjustedOffset = TargetElementUtil.adjustOffset(file, getDocument(), caretOffset);
 
-    PsiElement leaf = file.findElementAt(adjustedOffset);
-    PsiReference ref = file.findReferenceAt(adjustedOffset);
+    PsiElement leaf = file.findElementAt(TargetElementUtil.adjustOffset(file, getDocument(), caretOffset));
+    PsiReference ref = TargetElementUtil.findReference(editor);
 
     String expectedVariant = leaf == null ? null : myPolicy.getExpectedVariant(editor, file, leaf, ref);
     boolean prefixEqualsExpected = isPrefixEqualToExpectedVariant(caretOffset, leaf, ref, expectedVariant);