diff --git a/compiler/frontend/src/org/jetbrains/kotlin/psi/KtTypeCodeFragment.kt b/compiler/frontend/src/org/jetbrains/kotlin/psi/KtTypeCodeFragment.kt
index bf41e96d883..09e4d01edba 100644
--- a/compiler/frontend/src/org/jetbrains/kotlin/psi/KtTypeCodeFragment.kt
+++ b/compiler/frontend/src/org/jetbrains/kotlin/psi/KtTypeCodeFragment.kt
@@ -14,25 +14,20 @@
  * limitations under the License.
  */
 
-package org.jetbrains.kotlin.psi;
+package org.jetbrains.kotlin.psi
 
-import com.intellij.openapi.project.Project;
-import com.intellij.psi.PsiElement;
-import org.jetbrains.annotations.Nullable;
-import org.jetbrains.kotlin.KtNodeTypes;
+import com.intellij.openapi.project.Project
+import com.intellij.psi.PsiElement
+import org.jetbrains.kotlin.KtNodeTypes
 
-public class KtTypeCodeFragment extends KtCodeFragment {
-    public KtTypeCodeFragment(Project project, String name, CharSequence text, PsiElement context) {
-        super(project, name, text, null, KtNodeTypes.TYPE_CODE_FRAGMENT, context);
-    }
+class KtTypeCodeFragment(
+        project: Project,
+        name: String,
+        text: CharSequence,
+        context: PsiElement?
+) : KtCodeFragment(project, name, text, null, KtNodeTypes.TYPE_CODE_FRAGMENT, context) {
+    override fun getContentElement() = findChildByClass(KtTypeReference::class.java)
 
-    public boolean hasTypeReference() {
-        return getContentElement() != null;
-    }
-
-    @Nullable
-    @Override
-    public KtTypeReference getContentElement() {
-        return findChildByClass(KtTypeReference.class);
-    }
+    val hasTypeReference: Boolean
+        get() = getContentElement() != null
 }
diff --git a/idea/src/org/jetbrains/kotlin/idea/refactoring/changeSignature/ui/KotlinChangeSignatureDialog.kt b/idea/src/org/jetbrains/kotlin/idea/refactoring/changeSignature/ui/KotlinChangeSignatureDialog.kt
index 16bddee0393..3564edd5832 100644
--- a/idea/src/org/jetbrains/kotlin/idea/refactoring/changeSignature/ui/KotlinChangeSignatureDialog.kt
+++ b/idea/src/org/jetbrains/kotlin/idea/refactoring/changeSignature/ui/KotlinChangeSignatureDialog.kt
@@ -449,6 +449,6 @@ public class KotlinChangeSignatureDialog(
         }
 
         private fun hasTypeReference(codeFragment: PsiCodeFragment?): Boolean
-                = (codeFragment as? KtTypeCodeFragment)?.hasTypeReference() ?: false
+                = (codeFragment as? KtTypeCodeFragment)?.hasTypeReference ?: false
     }
 }