diff --git a/tests/models/modules/test_masked_autoencoder.py b/tests/models/modules/test_masked_autoencoder.py
index f96985a7b..42210bfdb 100644
--- a/tests/models/modules/test_masked_autoencoder.py
+++ b/tests/models/modules/test_masked_autoencoder.py
@@ -93,7 +93,7 @@ def test_forward_cuda(self):
 )
 class TestMAEDecoder(unittest.TestCase):
     def test_init(self):
-        return MAEDecoder(
+        MAEDecoder(
             seq_length=50,
             num_layers=2,
             num_heads=4,