diff --git a/setup.py b/setup.py
index bd7ac6e05..54a53d198 100644
--- a/setup.py
+++ b/setup.py
@@ -373,7 +373,7 @@ def check_cudnn_version_and_warn(global_option: str, required_cudnn_version: int
     if nvfuser_is_refactored:
         include_dirs.append(os.path.join(PYTORCH_HOME, "third_party/nvfuser/csrc"))
         library_dirs = nvfuser.__path__
-        extra_link_args.append("-lnvfuser")
+        extra_link_args.append("-lnvfuser_codegen")
     ext_modules.append(
         CUDAExtension(
             name='instance_norm_nvfuser_cuda',