Fix brace node extraction from class initializer

2013-06-05 16:03:51 +04:00
parent 62bdb37014
commit ef2af3a4e8
2 changed files with 7 additions and 5 deletions
@@ -44,10 +44,9 @@ public class JetClassInitializer extends JetDeclarationImpl implements JetStatem
        return body;
    }

-    @NotNull
+    @Nullable
    public PsiElement getOpenBraceNode() {
-        ASTNode openBraceNode = getNode().findChildByType(JetTokens.LBRACE);
-        assert openBraceNode != null;
-        return openBraceNode.getPsi();
+        JetExpression body = getBody();
+        return (body instanceof JetBlockExpression) ? ((JetBlockExpression) body).getLBrace() : null;
    }
 }