diff --git a/sdk/python/setup.py b/sdk/python/setup.py
index 0fc5984ad07..a6ae7f9b651 100644
--- a/sdk/python/setup.py
+++ b/sdk/python/setup.py
@@ -40,6 +40,15 @@
       'kfp_run',
       'kfp_run.api',
       'kfp_run.models',
+      'kfp_pipeline',
+      'kfp_pipeline.api',
+      'kfp_pipeline.models',
+      'kfp_uploadpipeline',
+      'kfp_uploadpipeline.api',
+      'kfp_uploadpipeline.models',
+      'kfp_job',
+      'kfp_job.api',
+      'kfp_job.models',
     ],
     include_package_data=True,
     entry_points = {