diff --git a/java/java-impl/src/com/intellij/refactoring/typeMigration/TypeMigrationLabeler.java b/java/java-impl/src/com/intellij/refactoring/typeMigration/TypeMigrationLabeler.java
index 1161f68490be..681e5a5a74af 100644
--- a/java/java-impl/src/com/intellij/refactoring/typeMigration/TypeMigrationLabeler.java
+++ b/java/java-impl/src/com/intellij/refactoring/typeMigration/TypeMigrationLabeler.java
@@ -468,9 +468,10 @@ public class TypeMigrationLabeler {
 
     if (!userDefinedType) {
       final Set<PsiTypeParameter> collector;
-      if (originalType instanceof PsiClassReferenceType) {
+      final PsiType rootInitialType = getElementType(getCurrentRoot().getElement());
+      if (rootInitialType  instanceof PsiClassReferenceType) {
         collector = new HashSet<PsiTypeParameter>();
-        final PsiJavaCodeReferenceElement reference = ((PsiClassReferenceType)originalType).getReference();
+        final PsiJavaCodeReferenceElement reference = ((PsiClassReferenceType)rootInitialType).getReference();
         RefactoringUtil.collectTypeParameters(collector, reference);
       } else {
         collector = Collections.emptySet();