diff --git a/tests/python/unittest/test_gluon.py b/tests/python/unittest/test_gluon.py
index 8542722090c9..59eb65444152 100644
--- a/tests/python/unittest/test_gluon.py
+++ b/tests/python/unittest/test_gluon.py
@@ -473,7 +473,7 @@ def hybrid_forward(self, F, a, b):
         foo1 = FooNested(prefix='foo_nested_hybridized')
         foo1.hybridize()
         foo2 = FooNested(prefix='foo_nested_nohybrid')
-        for _ in range(2):
+        for _ in range(2): # Loop for 2 times to trigger forwarding of the cached version
             out1 = foo1(*arg_inputs)
             out2 = foo2(*arg_inputs)
             if isinstance(out1, tuple):