diff --git a/enclave_shelly/__init__.py b/enclave_shelly/__init__.py
index 4e75b6b..db58be9 100644
--- a/enclave_shelly/__init__.py
+++ b/enclave_shelly/__init__.py
@@ -1,3 +1,3 @@
-__all__ = ["shelly_base"]
+__all__ = ["ShellyBase"]
 
 from .shelly_base import ShellyBase
diff --git a/pyproject.toml b/pyproject.toml
index 5f94b6e..77bbeee 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -7,6 +7,9 @@ name = "enclave_shelly"
 version = "1.0.0"
 dependencies = ["aiohttp"]
 
+[tool.setuptools]
+packages = ["enclave_shelly"]
+
 [tool.setuptools.packages.find]
 where = ["."]
 namespaces = false