From 5117bf81186dc410c1347b613dd037fdb8985033 Mon Sep 17 00:00:00 2001
From: Nikolay Krasko <nikolay.krasko@jetbrains.com>
Date: Fri, 9 Nov 2012 20:59:01 +0400
Subject: [PATCH] Don't postpone assigning created body

---
 .../codeInsight/OverrideImplementMethodsHandler.java  | 11 ++++-------
 1 file changed, 4 insertions(+), 7 deletions(-)

diff --git a/idea/src/org/jetbrains/jet/plugin/codeInsight/OverrideImplementMethodsHandler.java b/idea/src/org/jetbrains/jet/plugin/codeInsight/OverrideImplementMethodsHandler.java
index 781abd15869..8613342c4ba 100644
--- a/idea/src/org/jetbrains/jet/plugin/codeInsight/OverrideImplementMethodsHandler.java
+++ b/idea/src/org/jetbrains/jet/plugin/codeInsight/OverrideImplementMethodsHandler.java
@@ -69,9 +69,10 @@ public abstract class OverrideImplementMethodsHandler implements LanguageCodeIns
             List<DescriptorClassMember> selectedElements
     ) {
         JetClassBody body = classOrObject.getBody();
-        Project project = classOrObject.getProject();
         if (body == null) {
-            body = JetPsiFactory.createEmptyClassBody(project);
+            Project project = classOrObject.getProject();
+            classOrObject.add(JetPsiFactory.createWhiteSpace(project));
+            body = (JetClassBody) classOrObject.add(JetPsiFactory.createEmptyClassBody(project));
         }
 
         PsiElement afterAnchor = findInsertAfterAnchor(editor, body);
@@ -87,12 +88,8 @@ public abstract class OverrideImplementMethodsHandler implements LanguageCodeIns
             afterAnchor = added;
             elementsToCompact.add((JetElement) added);
         }
-        ReferenceToClassesShortening.compactReferenceToClasses(elementsToCompact);
 
-        if (!body.isPhysical()) {
-            classOrObject.add(JetPsiFactory.createWhiteSpace(project));
-            classOrObject.add(body);
-        }
+        ReferenceToClassesShortening.compactReferenceToClasses(elementsToCompact);
     }
 
     @Nullable