diff --git a/setup.py b/setup.py
index f926f86..3371364 100644
--- a/setup.py
+++ b/setup.py
@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='roboimi',
-    packages=[package for package in find_packages() if package.startswith("robosuite")],
+    packages=[package for package in find_packages() if package.startswith("roboimi")],
     version='0.0.2',
     author="Zongdao Li, Haoran Zhou, Yichao Huang, Yuhan Zhao, Yang Lu",
     author_email="leeeezd0016@outlook.com, jou072@126.com",