diff --git a/setup.py b/setup.py
index a5d91fc..c925740 100644
--- a/setup.py
+++ b/setup.py
@@ -5,14 +5,14 @@
 
 setuptools.setup(
     name="ast_scope",  # Replace with your own username
-    version="0.5.0",
+    version="0.5.1",
     author="Kavi Gupta",
     author_email="ast_scope@kavigupta.org",
     description="Annotates a Python AST with the scope of symbols.",
     long_description=long_description,
     long_description_content_type="text/markdown",
     url="https://github.com/kavigupta/ast_scope",
-    packages=setuptools.find_packages("ast_scope"),
+    packages=setuptools.find_packages(include=["ast_scope"]),
     classifiers=[
         "Programming Language :: Python :: 3.9",
         "Programming Language :: Python :: 3.10",