diff --git a/pyproject.toml b/pyproject.toml
index e02cddc..c08f224 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -52,7 +52,10 @@ classifiers=[
 dependencies = ["requests>=2.31.0"]
 
 [tool.setuptools]
-packages = ["shellhub"]
+packages = [
+    "shellhub",
+    "shellhub.models",
+]
 
 [tool.setuptools.dynamic]
 version = {attr = "shellhub.__version__"}
diff --git a/shellhub/__init__.py b/shellhub/__init__.py
index 0222d04..6915165 100644
--- a/shellhub/__init__.py
+++ b/shellhub/__init__.py
@@ -1,5 +1,5 @@
 # Increment versions here according to SemVer
-__version__ = "0.2.0"
+__version__ = "0.2.1"
 
 from .models.device import ShellHubDevice, ShellHubDeviceInfo
 from .models.base import ShellHub