diff --git a/MANIFEST.in b/MANIFEST.in
index 096dd50..60e8985 100644
--- a/MANIFEST.in
+++ b/MANIFEST.in
@@ -1,2 +1,3 @@
 include LICENSE
 include requirements.txt
+include aioesphomeapi/py.typed
diff --git a/aioesphomeapi/client.py b/aioesphomeapi/client.py
index 293fa29..2f597b9 100644
--- a/aioesphomeapi/client.py
+++ b/aioesphomeapi/client.py
@@ -114,7 +114,7 @@ class APIClient:
         eventloop: asyncio.AbstractEventLoop,
         address: str,
         port: int,
-        password: str,
+        password: Optional[str],
         *,
         client_info: str = "aioesphomeapi",
         keepalive: float = 15.0,
diff --git a/aioesphomeapi/model.py b/aioesphomeapi/model.py
index 803b4dc..2d89229 100644
--- a/aioesphomeapi/model.py
+++ b/aioesphomeapi/model.py
@@ -423,7 +423,7 @@ class NumberState(EntityState):
     missing_state: bool = False
 
 
-COMPONENT_TYPE_TO_INFO = {
+COMPONENT_TYPE_TO_INFO: Dict[str, Type[EntityInfo]] = {
     "binary_sensor": BinarySensorInfo,
     "cover": CoverInfo,
     "fan": FanInfo,
diff --git a/aioesphomeapi/py.typed b/aioesphomeapi/py.typed
new file mode 100644
index 0000000..e69de29