diff --git a/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetClassBody.java b/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetClassBody.java
index 965c08abdc7..6e3683215b4 100644
--- a/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetClassBody.java
+++ b/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetClassBody.java
@@ -65,7 +65,7 @@ public class JetClassBody extends JetElementImplStub<PsiJetPlaceHolderStub<JetCl
 
     @Nullable
     public JetClassObject getClassObject() {
-        return (JetClassObject) findChildByType(JetNodeTypes.CLASS_OBJECT);
+        return getStubOrPsiChild(JetStubElementTypes.CLASS_OBJECT);
     }
 
     @NotNull