diff --git a/test/completion/classes.py b/test/completion/classes.py
index 5ebc6471..929f9665 100644
--- a/test/completion/classes.py
+++ b/test/completion/classes.py
@@ -544,3 +544,23 @@ class TestSuper(Super):
         def a():
             #? 
             super()
+
+
+# -----------------
+# if flow at class level
+# -----------------
+class TestX(object):
+    def normal_method(self):
+        return 1
+
+    if True:
+        def conditional_method(self):
+            var = self.normal_method()
+            #? int()
+            var
+            return 2
+
+    def other_method(self):
+        var = self.conditional_method()
+        #? int()
+        var