diff --git a/test/integration/test_integration.py b/test/integration/test_integration.py
index 0b69882bb..7246bf02f 100644
--- a/test/integration/test_integration.py
+++ b/test/integration/test_integration.py
@@ -73,7 +73,7 @@
     AQInt8WeightOnlyQuantizedLinearWeight3,
     AutoQuantizableLinearWeight,
     AQFloat8WeightOnlyQuantizedLinearWeight,
-    AQFloat8DynamicallyQuantizedLinearWeight,
+    AQFloat8PerRowScalingDynamicallyQuantizedLinearWeight,
 )
 from torch.ao.quantization.quantize_fx import convert_to_reference_fx, prepare_fx
 import os