From a9d18c0cfacddb988a850f9491786526d85bdb23 Mon Sep 17 00:00:00 2001
From: Dmitry Trofimov <dmitry.trofimov@jetbrains.com>
Date: Mon, 7 Jan 2013 19:01:50 +0100
Subject: [PATCH] EA-42015

---
 .../python/documentation/PyDocstringGenerator.java          | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/python/src/com/jetbrains/python/documentation/PyDocstringGenerator.java b/python/src/com/jetbrains/python/documentation/PyDocstringGenerator.java
index ed4c46793537..3275bd13102f 100644
--- a/python/src/com/jetbrains/python/documentation/PyDocstringGenerator.java
+++ b/python/src/com/jetbrains/python/documentation/PyDocstringGenerator.java
@@ -330,7 +330,11 @@ public class PyDocstringGenerator {
       if (myFunction != null) {
         myFunction = CodeInsightUtilBase.forcePsiPostprocessAndRestoreElement(myFunction);
       }
-      myDocStringOwner = CodeInsightUtilBase.forcePsiPostprocessAndRestoreElement(myDocStringOwner);
+      PyDocStringOwner owner = CodeInsightUtilBase.forcePsiPostprocessAndRestoreElement(myDocStringOwner);
+      if (owner != null) {
+        myDocStringOwner = owner;
+      }
+
       myDocStringExpression = myDocStringOwner.getDocStringExpression();
     }
     else {