diff --git a/python/pyproject.toml b/python/pyproject.toml
index c39c65f6e133..c3ff9fdf7c63 100755
--- a/python/pyproject.toml
+++ b/python/pyproject.toml
@@ -131,6 +131,9 @@ sglang = "sglang.cli.main:main"
   "srt/mem_cache/storage/hf3fs/hf3fs_utils.cpp",
   "srt/speculative/cpp_ngram/*.cpp",
   "srt/speculative/cpp_ngram/*.h",
+  "jit_kernel/include/sgl_kernel/*.h",
+  "jit_kernel/include/sgl_kernel/*.cuh",
+  "jit_kernel/csrc/*.cuh"
 ]
 
 [tool.setuptools.packages.find]