diff --git a/aw_client/__init__.py b/aw_client/__init__.py
index 099726e..d43dcb9 100644
--- a/aw_client/__init__.py
+++ b/aw_client/__init__.py
@@ -1 +1,3 @@
-from .client import ActivityWatchClient  # noqa
+from .client import ActivityWatchClient
+
+__all__ = ["ActivityWatchClient"]