diff --git a/setup.py b/setup.py
index 37fae39..f8b91f6 100644
--- a/setup.py
+++ b/setup.py
@@ -21,7 +21,7 @@ setup(
     version=get_version(),
     author="Dalibo",
     author_email="contact@dalibo.com",
-    packages=find_packages("."),
+    packages=find_packages(include=['check_patroni*']),
     include_package_data=True,
     url="https://github.com/dalibo/check_patroni",
     license="PostgreSQL",
@@ -38,7 +38,6 @@ setup(
     keywords="patroni nagios check",
     python_requires=">=3.6",
     install_requires=[
-        "wheel",
         "attrs >= 17, !=21.1",
         "urllib3 >= 1.26.6",
         "nagiosplugin >= 1.3.2",
@@ -46,6 +45,7 @@ setup(
     ],
     extras_require={
         "dev": [
+            "wheel",
             "black",
             "check-manifest",
             "flake8",