diff --git a/tests/ngc_mmar_loading.py b/tests/ngc_mmar_loading.py
index 8aa3859d87..c1ed22de5d 100644
--- a/tests/ngc_mmar_loading.py
+++ b/tests/ngc_mmar_loading.py
@@ -27,7 +27,7 @@ def setUp(self):
     @parameterized.expand((item,) for item in MODEL_DESC)
     def test_loading_mmar(self, item):
         pretrained_model = load_from_mmar(item=item, mmar_dir="./", map_location="cpu")
-        self.assertTrue(isinstance(pretrained_model, torch.nn.Modules))
+        self.assertTrue(isinstance(pretrained_model, torch.nn.Module))
 
     def tearDown(self):
         print(os.listdir(self.test_dir))