diff --git a/setup.py b/setup.py
index 6280761..25b03bf 100644
--- a/setup.py
+++ b/setup.py
@@ -18,7 +18,7 @@ def get_requirements():
 
 def get_version():
     current_dir = os.path.abspath(os.path.dirname(__file__))
-    version_file = os.path.join(current_dir, "cb_loss", "__init__.py")
+    version_file = os.path.join(current_dir, "cbloss", "__init__.py")
     with io.open(version_file, encoding="utf-8") as f:
         return re.search(r'^__version__ = [\'"]([^\'"]*)[\'"]', f.read(), re.M).group(1)