1 files changed, 2 insertions, 2 deletions
diff --git a/python/src/com/jetbrains/python/hierarchy/PyTypeHierarchyBrowser.java b/python/src/com/jetbrains/python/hierarchy/PyTypeHierarchyBrowser.java
index 643440571618..817d462044fe 100644
--- a/python/src/com/jetbrains/python/hierarchy/PyTypeHierarchyBrowser.java
+++ b/python/src/com/jetbrains/python/hierarchy/PyTypeHierarchyBrowser.java
@@ -52,10 +52,10 @@ public class PyTypeHierarchyBrowser extends TypeHierarchyBrowserBase {
 
   @Nullable
   protected PsiElement getElementFromDescriptor(@NotNull HierarchyNodeDescriptor descriptor) {
-    if (!(descriptor instanceof PyTypeHierarchyNodeDescriptor)) {
+    if (!(descriptor instanceof PyHierarchyNodeDescriptor)) {
       return null;
     }
-    return ((PyTypeHierarchyNodeDescriptor)descriptor).getClassElement();
+    return ((PyHierarchyNodeDescriptor)descriptor).getPsiElement();
   }
 
   protected void createTrees(@NotNull Map<String, JTree> trees) {