diff --git a/python/setup.py b/python/setup.py
index a0297d4f9dcf..4d14dfd3cb97 100755
--- a/python/setup.py
+++ b/python/setup.py
@@ -235,6 +235,7 @@ def run(self):
             "pyspark",
             "pyspark.cloudpickle",
             "pyspark.mllib",
+            "pyspark.ml.connect",
             "pyspark.mllib.linalg",
             "pyspark.mllib.stat",
             "pyspark.ml",
@@ -245,7 +246,10 @@ def run(self):
             "pyspark.sql",
             "pyspark.sql.avro",
             "pyspark.sql.connect",
+            "pyspark.sql.connect.avro",
+            "pyspark.sql.connect.client",
             "pyspark.sql.connect.proto",
+            "pyspark.sql.connect.streaming",
             "pyspark.sql.pandas",
             "pyspark.sql.protobuf",
             "pyspark.sql.streaming",