diff --git a/README.md b/README.md
index 4217894..c5ab772 100644
--- a/README.md
+++ b/README.md
@@ -25,7 +25,7 @@ pip install alias-free-torch==0.0.6 phaseaug
 ```
 - Insert PhaseAug in your code, check [train.py](./train.py) as a example.
 ```python
-from phaseaug import PhaseAug
+from phaseaug.phaseaug import PhaseAug
 ...
 # define phaseaug
 aug = PhaseAug()
diff --git a/phaseaug/__init__.py b/phaseaug/__init__.py
index c074933..27cece6 100644
--- a/phaseaug/__init__.py
+++ b/phaseaug/__init__.py
@@ -1 +1 @@
-from phaseaug import *
\ No newline at end of file
+from . import phaseaug
diff --git a/setup.py b/setup.py
index d869c24..b49f25d 100644
--- a/setup.py
+++ b/setup.py
@@ -6,16 +6,15 @@
 
 setup(
     name                = 'phaseaug',
-    version             = '0.0.1',
+    version             = '0.0.2',
     description         = 'PhaseAug: A Differentiable Augmentation for Speech Synthesis to Simulate One-to-Many Mapping',
     long_description=long_description,
     long_description_content_type="text/markdown",
     author              = 'junjun3518',
     author_email        = 'junjun3518@gmail.com',
     url                 = 'https://github.com/mindslab-ai/phaseaug',
-    install_requires    =  ['torch', 'alias-free-torch'],
-    packages            = find_packages(where = "phaseaug"),
-    package_dir         = {"": "phaseaug"},
+    install_requires    = ['torch', 'alias-free-torch'],
+    packages            = ['phaseaug'],
     keywords            = ['torch','pytorch','augmentation', 'diffaugment', 'speech synthesis', 'vocoder'],
     python_requires     = '>=3',
     zip_safe            = False,