diff --git a/setup.py b/setup.py
index c9deeca..d23e497 100644
--- a/setup.py
+++ b/setup.py
@@ -49,9 +49,9 @@
     author='Deepak Yadav', 
     author_email='dky.united@gmail.com', 
     license='MIT', 
-    packages=find_packages(),  # Include all packages in the new version
+    packages=find_packages('robustbase'),  # Include all packages in the new version
     include_package_data=True,
-    package_dir={'robustbase': 'robustbase'},
+    package_dir={'': 'robustbase'},
     project_urls={
         'Source': 'https://github.com/deepak7376/robustbase',
         'Tracker': 'https://github.com/deepak7376/robustbase/issues',