diff --git a/compass/db/api/cluster.py b/compass/db/api/cluster.py
index bb9805cd..d82f9514 100644
--- a/compass/db/api/cluster.py
+++ b/compass/db/api/cluster.py
@@ -2165,7 +2165,7 @@ def update_clusterhost_state_internal(
     clusterhost = _get_clusterhost(clusterhost_id, session=session)
     return _update_clusterhost_state(
         clusterhost, from_database_only=from_database_only,
-        session=None, **kwargs
+        session=session, **kwargs
     )