diff --git a/custom_components/google_geocode/sensor.py b/custom_components/google_geocode/sensor.py
index 09bb0cc..d9a9428 100644
--- a/custom_components/google_geocode/sensor.py
+++ b/custom_components/google_geocode/sensor.py
@@ -184,7 +184,7 @@ def update(self):
         if zone_check == self._zone_check_current and zone_check != 'not_home':    
             return
 
-        self._zone_check_current = self.hass.states.get(self._origin_entity_id).state
+        self._zone_check_current = zone_check
         lat = self._origin
         current = lat
         self._reset_attributes()