diff --git a/setup.py b/setup.py
index 35df839..f1e98be 100644
--- a/setup.py
+++ b/setup.py
@@ -189,6 +189,7 @@ def get_extra_link_args():
         CUDAExtension(
             "awq_v2_ext",
             [
+                "awq_ext/pybind_awq_v2.cpp",
                 "awq_ext/quantization_new/gemv/gemv_cuda.cu",
                 "awq_ext/quantization_new/gemm/gemm_cuda.cu",
             ],