diff --git a/halo2_gadgets/src/utilities/lookup_range_check.rs b/halo2_gadgets/src/utilities/lookup_range_check.rs
index a1d9b41ad..bfc270045 100644
--- a/halo2_gadgets/src/utilities/lookup_range_check.rs
+++ b/halo2_gadgets/src/utilities/lookup_range_check.rs
@@ -994,10 +994,11 @@ mod tests {
             fn configure(meta: &mut ConstraintSystem<F>) -> Self::Config {
                 let running_sum = meta.advice_column();
                 let table_idx = meta.lookup_table_column();
+                let table_range_check_tag = meta.lookup_table_column();
                 let constants = meta.fixed_column();
                 meta.enable_constant(constants);
 
-                LookupRangeCheckConfigOptimized::<F, K>::configure(meta, running_sum, table_idx)
+                LookupRangeCheckConfigOptimized::<F, K>::configure_with_tag(meta, running_sum, table_idx, table_range_check_tag)
             }
 
             fn synthesize(
@@ -1093,10 +1094,11 @@ mod tests {
             fn configure(meta: &mut ConstraintSystem<F>) -> Self::Config {
                 let running_sum = meta.advice_column();
                 let table_idx = meta.lookup_table_column();
+                let table_range_check_tag = meta.lookup_table_column();
                 let constants = meta.fixed_column();
                 meta.enable_constant(constants);
 
-                LookupRangeCheckConfigOptimized::<F, K>::configure(meta, running_sum, table_idx)
+                LookupRangeCheckConfigOptimized::<F, K>::configure_with_tag(meta, running_sum, table_idx, table_range_check_tag)
             }
 
             fn synthesize(