diff --git a/python/testData/inspections/AddMethodFromMethod.py b/python/testData/inspections/AddMethodFromMethod.py
index 84cb7101c4f0..26f30bc3ada1 100644
--- a/python/testData/inspections/AddMethodFromMethod.py
+++ b/python/testData/inspections/AddMethodFromMethod.py
@@ -4,3 +4,8 @@ class A:
 
     def foo(self, a):
         self.<caret><warning descr="Unresolved attribute reference 'y' for class 'A'">y</warning>(1, a)
+
+# Some comment
+
+class B:
+    pass
diff --git a/python/testData/inspections/AddMethodFromMethod_after.py b/python/testData/inspections/AddMethodFromMethod_after.py
index 591842381d86..978acaf15275 100644
--- a/python/testData/inspections/AddMethodFromMethod_after.py
+++ b/python/testData/inspections/AddMethodFromMethod_after.py
@@ -7,3 +7,8 @@ class A:
 
     def y(self, param, a):
         pass
+
+# Some comment
+
+class B:
+    pass