diff --git a/tests/test_models/test_backbone.py b/tests/test_models/test_backbone.py
index 36fdd78fc..084849aa4 100644
--- a/tests/test_models/test_backbone.py
+++ b/tests/test_models/test_backbone.py
@@ -676,13 +676,13 @@ def test_fastscnn_backbone():
     model.init_weights()
     model.train()
     batch_size = 4
-    imgs = torch.randn(batch_size, 3, 1024, 2048)
+    imgs = torch.randn(batch_size, 3, 512, 1024)
     feat = model(imgs)
 
     assert len(feat) == 3
     # higher-res
-    assert feat[0].shape == torch.Size([batch_size, 64, 128, 256])
+    assert feat[0].shape == torch.Size([batch_size, 64, 64, 128])
     # lower-res
-    assert feat[1].shape == torch.Size([batch_size, 128, 32, 64])
+    assert feat[1].shape == torch.Size([batch_size, 128, 16, 32])
     # FFM output
-    assert feat[2].shape == torch.Size([batch_size, 128, 128, 256])
+    assert feat[2].shape == torch.Size([batch_size, 128, 64, 128])
diff --git a/tests/test_ops/test_inverted_residual_module.py b/tests/test_ops/test_inverted_residual_module.py
index 0e3d876e3..41c663b99 100644
--- a/tests/test_ops/test_inverted_residual_module.py
+++ b/tests/test_ops/test_inverted_residual_module.py
@@ -10,7 +10,7 @@ def test_inv_residual():
         # test stride assertion.
         InvertedResidual(32, 32, 3, 4)
 
-    # test default config without res connection.
+    # test default config with res connection.
     # set expand_ratio = 4, stride = 1 and inp=oup.
     inv_module = InvertedResidual(32, 32, 1, 4)
     assert inv_module.use_res_connect