diff --git a/sgl-kernel/setup.py b/sgl-kernel/setup.py
index 7b21ba59390..807f9404c00 100644
--- a/sgl-kernel/setup.py
+++ b/sgl-kernel/setup.py
@@ -235,6 +235,12 @@ def copy_deepgemm_to_build_lib(self):
     ),
 ]
 
+
+def get_py_ver():
+    major, minor = sys.version_info[:2]
+    return f"cp{major}{minor}"
+
+
 setup(
     name="sgl-kernel",
     version=_get_version(),
@@ -245,5 +251,5 @@ def copy_deepgemm_to_build_lib(self):
         "build_ext": BuildExtension.with_options(use_ninja=True),
         "build_py": CustomBuildPy,
     },
-    options={"bdist_wheel": {"py_limited_api": "cp39"}},
+    options={"bdist_wheel": {"py_limited_api": get_py_ver()}},
 )