diff --git a/src/brevitas_examples/imagenet_classification/ptq/ptq_common.py b/src/brevitas_examples/imagenet_classification/ptq/ptq_common.py
index a45f89367..95153bd0e 100644
--- a/src/brevitas_examples/imagenet_classification/ptq/ptq_common.py
+++ b/src/brevitas_examples/imagenet_classification/ptq/ptq_common.py
@@ -120,7 +120,7 @@
                     'sym': Fp8e4m3ActPerTensorFloat}},
             'mse': {
                 'per_tensor': {
-                    'sym': Fp8e4m3ActPerTensorFloat},}}}}
+                    'sym': Fp8e4m3ActPerTensorFloatMSE}}}}}
 
 
 def quantize_model(