diff --git a/test/prototype/test_awq.py b/test/prototype/test_awq.py
index d5318de7ca..22a6d93313 100644
--- a/test/prototype/test_awq.py
+++ b/test/prototype/test_awq.py
@@ -59,8 +59,12 @@ def forward(self, x):
 device_to_base_configs = {
     "cuda": [
         Int4WeightOnlyConfig(group_size=128),
-        # Note: the functionality unit test doesn't work for hqq
         Int4WeightOnlyConfig(group_size=128, int4_packing_format="tile_packed_to_4d"),
+        Int4WeightOnlyConfig(
+            group_size=128,
+            int4_packing_format="tile_packed_to_4d",
+            int4_choose_qparams_algorithm="hqq",
+        ),
     ],
     "cpu": [Int4WeightOnlyConfig(group_size=128, int4_packing_format="opaque")],
     "xpu": [Int4WeightOnlyConfig(group_size=128, int4_packing_format="plain_int32")],