diff --git a/chromadb/api/rust.py b/chromadb/api/rust.py
index e4885072088..29c5abfadb0 100644
--- a/chromadb/api/rust.py
+++ b/chromadb/api/rust.py
@@ -74,8 +74,7 @@ def get_tenant(self, name: str) -> Tenant:
 
     @override
     def heartbeat(self) -> int:
-        # TODO(hammadb): the precommit hooks don't know about the .pyi file
-        return self.bindings.heartbeat()  # type: ignore
+        return self.bindings.heartbeat()
 
     @override
     def count_collections(
diff --git a/pyproject.toml b/pyproject.toml
index fdc51b199cb..dc9656f66ce 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -57,6 +57,7 @@ asyncio_mode = "auto"
 
 [tool.mypy]
 ignore_errors = false
+mypy_path = "rust/python_bindings"
 
 [[tool.mypy.overrides]]
 module = ["chromadb.proto.*"]