diff --git a/setup.py b/setup.py
index 573bfba..96a5e45 100644
--- a/setup.py
+++ b/setup.py
@@ -44,7 +44,7 @@ def read_requirements(path):
 
 # loading version from setup.py
 with codecs.open(
-    os.path.join(here, "einstein-ait-prod/__init__.py"), encoding="utf-8"
+    os.path.join(here, "prompting/__init__.py"), encoding="utf-8"
 ) as init_file:
     version_match = re.search(
         r"^__version__ = ['\"]([^'\"]*)['\"]", init_file.read(), re.M