diff --git a/setup.py b/setup.py
index 8b4be3b..6986676 100644
--- a/setup.py
+++ b/setup.py
@@ -14,7 +14,7 @@ setuptools.setup(
     url="https://git.rudefox.io/bj/torch-agent",
     packages=setuptools.find_packages(),
     entry_points = {
-        'console_scripts': ['torch-agent=torch_agent:main'],
+        'console_scripts': ['torch-agent=torch_agent.torch_agent:main'],
     },
     classifiers=[
         "Programming Language :: Python :: 3",