diff --git a/pyproject.toml b/pyproject.toml
index fc20e7734..d993e24f0 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -50,6 +50,7 @@ packages = [
     "flashinfer.jit.attention",
     "flashinfer.triton",
     "flashinfer.triton.kernels",
+    "flashinfer.comm",
     "flashinfer.cudnn",
 ]
 include-package-data = false