diff --git a/src/main/kotlin/org/nextras/orm/intellij/reference/ModifierClassName.kt b/src/main/kotlin/org/nextras/orm/intellij/reference/ModifierClassName.kt
index fc0ead2..5086cd5 100644
--- a/src/main/kotlin/org/nextras/orm/intellij/reference/ModifierClassName.kt
+++ b/src/main/kotlin/org/nextras/orm/intellij/reference/ModifierClassName.kt
@@ -6,7 +6,6 @@ import com.intellij.psi.PsiPolyVariantReferenceBase
 import com.intellij.psi.ResolveResult
 import com.jetbrains.php.PhpIndex
 import com.jetbrains.php.lang.psi.resolve.types.PhpType
-import org.nextras.orm.intellij.utils.OrmUtils
 import org.nextras.orm.intellij.utils.PhpIndexUtils
 
 class ModifierClassName(
@@ -17,7 +16,6 @@ class ModifierClassName(
 		val phpIndex = PhpIndex.getInstance(this.element.project)
 		val result = PhpIndexUtils.getByType(PhpType().add(fqnClass), phpIndex)
 		return result
-			.filter { OrmUtils.OrmClass.ENTITY.`is`(it, phpIndex) }
 			.map {
 				object : ResolveResult {
 					override fun getElement(): PsiElement {
diff --git a/src/main/kotlin/org/nextras/orm/intellij/reference/ModifierClassNameProvider.kt b/src/main/kotlin/org/nextras/orm/intellij/reference/ModifierClassNameProvider.kt
index 714ece0..4f07988 100644
--- a/src/main/kotlin/org/nextras/orm/intellij/reference/ModifierClassNameProvider.kt
+++ b/src/main/kotlin/org/nextras/orm/intellij/reference/ModifierClassNameProvider.kt
@@ -41,6 +41,7 @@ class ModifierClassNameProvider : PsiReferenceProvider() {
 							PlatformPatterns.psiElement(PhpDocTokenTypes.DOC_IDENTIFIER).withText("1")
 						)
 					),
+					PlatformPatterns.psiElement(PhpDocTokenTypes.DOC_IDENTIFIER).withText("wrapper"),
 				)
 			)
 			.withLanguage(PhpLanguage.INSTANCE)