diff --git a/jedi/api/helpers.py b/jedi/api/helpers.py
index 09659f56..e1e9e8a6 100644
--- a/jedi/api/helpers.py
+++ b/jedi/api/helpers.py
@@ -97,7 +97,11 @@ def _get_code_for_stack(code_lines, module_node, position):
         if leaf == ';':
             user_stmt = leaf.parent
         else:
-            user_stmt = leaf.get_definition()
+            user_stmt = leaf
+            while True:
+                if user_stmt.parent.type in ('file_input', 'suite'):
+                    break
+                user_stmt = user_stmt.parent
 
         if user_stmt.parent.type == 'simple_stmt':
             user_stmt = user_stmt.parent