diff --git a/llmc/compression/quantization/smoothquant.py b/llmc/compression/quantization/smoothquant.py
index 71ae72612..061b65068 100644
--- a/llmc/compression/quantization/smoothquant.py
+++ b/llmc/compression/quantization/smoothquant.py
@@ -71,7 +71,7 @@ def subset_transform(
         inspect_module,
         subset_kwargs,
     ):
-        if not self.filter_subset(prev_op):
+        if not self.filter_subset(layers_dict, prev_op):
             logger.info('Do not transform this subset.')
             return
         layers = list(layers_dict.values())