diff --git a/quantize/gptq/quant.py b/quantize/gptq/quant.py
index 897289ff..23933bae 100644
--- a/quantize/gptq/quant.py
+++ b/quantize/gptq/quant.py
@@ -148,7 +148,7 @@ def make_quant(module, names, bits, groupsize, name=''):
         make_quant(child, names, bits, groupsize, name + '.' + name1 if name != '' else name1)
 
 def make_quant_custom(module, names, bits, groupsize, name=''):
-    if isinstance(module, QuantLinear):
+    if isinstance(module, QuantLinear_custom):
         return
     for attr in dir(module):
         tmp = getattr(module, attr)