1 files changed, 5 insertions, 2 deletions
diff --git a/tests/extensions/test_comparetozero.py b/tests/extensions/test_comparetozero.py
index 52be28c1f..7b5bb1624 100644
--- a/tests/extensions/test_comparetozero.py
+++ b/tests/extensions/test_comparetozero.py
@@ -15,19 +15,22 @@
 
 import os
 import unittest
-from typing import Optional
+from typing import TYPE_CHECKING, Optional
 
 from pylint import checkers
 from pylint.extensions.comparetozero import CompareToZeroChecker
 from pylint.lint import PyLinter
 from pylint.reporters import BaseReporter
 
+if TYPE_CHECKING:
+    from pylint.reporters.ureports.nodes import Section
+
 
 class CompareToZeroTestReporter(BaseReporter):
     def on_set_current_module(self, module: str, filepath: Optional[str]) -> None:
         self.messages = []
 
-    def _display(self, layout):
+    def _display(self, layout: "Section") -> None:
         pass