diff --git a/tests/sparsezoo/models/classification/test_resnet.py b/tests/sparsezoo/models/classification/test_resnet.py
index 45dad226..1c6e7f29 100644
--- a/tests/sparsezoo/models/classification/test_resnet.py
+++ b/tests/sparsezoo/models/classification/test_resnet.py
@@ -139,10 +139,10 @@ def test_resnet_50(
             True,
             "pytorch",
             "sparseml",
-            "imagenet",
+            "imagenet-augmented",
             None,
             "pruned_quant",
-            "conservative",
+            "aggressive",
             None,
         ),
         (True, "tensorflow_v1", "sparseml", "imagenet", None, "base", "none", None),
@@ -157,7 +157,6 @@ def test_resnet_50(
             "conservative",
             None,
         ),
-        (True, "pytorch", "sparseml", "imagenette", None, "pruned", "moderate", None),
         (
             True,
             "pytorch",