diff --git a/setup.py b/setup.py
index 0963bf4..41b0ca4 100644
--- a/setup.py
+++ b/setup.py
@@ -49,5 +49,5 @@
     # simple. Or you can use find_packages().
     # TODO: IF LIBRARY FILES ARE A PACKAGE FOLDER,
     #       CHANGE `py_modules=['...']` TO `packages=['...']`
-    packages=["adafruit_display_button"],
+    py_modules=["adafruit_button"],
 )