diff --git a/tests/networks/nets/test_restormer.py b/tests/networks/nets/test_restormer.py
index 9b54b7a765..f0466caa2d 100644
--- a/tests/networks/nets/test_restormer.py
+++ b/tests/networks/nets/test_restormer.py
@@ -40,10 +40,10 @@
 ]
 
 RESTORMER_CONFIGS = [
-    # 2-level architecture
+    # 2-level architecture test
     {"num_blocks": [1, 1], "heads": [1, 1]},
     {"num_blocks": [2, 1], "heads": [2, 1]},
-    # 3-level architecture
+    # 3-level architecture test
     {"num_blocks": [1, 1, 1], "heads": [1, 1, 1]},
     {"num_blocks": [2, 1, 1], "heads": [2, 1, 1]},
 ]