diff --git a/tests/sparseml/export/transformers/test_generative_transformers.py b/tests/sparseml/export/transformers/test_generative_transformers.py
index 1a9af341cb9..d09b5e520d8 100644
--- a/tests/sparseml/export/transformers/test_generative_transformers.py
+++ b/tests/sparseml/export/transformers/test_generative_transformers.py
@@ -131,7 +131,7 @@ def test_export_with_recipe(self, setup):
 
         recipe = """test_stage:
           quant_modifiers:
-            QuantizationModifier:
+            LegacyQuantizationModifier:
               post_oneshot_calibration: False
               scheme_overrides:
                 Embedding: