diff --git a/src/main/kotlin/de/sirywell/methodhandleplugin/type/Type.kt b/src/main/kotlin/de/sirywell/methodhandleplugin/type/Type.kt
index 2da90a0..0eaf475 100644
--- a/src/main/kotlin/de/sirywell/methodhandleplugin/type/Type.kt
+++ b/src/main/kotlin/de/sirywell/methodhandleplugin/type/Type.kt
@@ -44,13 +44,8 @@ data class DirectType(val psiType: PsiType) : Type {
         return when (other) {
             is BotType -> this
             is TopType -> other
-            is DirectType -> {
-                if (other.psiType == psiType) {
-                    return this
-                } else {
-                    return TopType
-                }
-            }
+            is DirectType -> if (other.psiType == psiType) this else TopType
+
         }
     }