diff --git a/api.php b/api.php
index be42c39..8007dd7 100644
--- a/api.php
+++ b/api.php
@@ -112,7 +112,7 @@ if ($_SERVER['REQUEST_METHOD'] === 'GET') {
     }
 
     if (isset($_GET['pokeCount'])) {
-        $state = custom_ha_state('main', $config['entities']['poke_count']);
+        $state = custom_ha_state('mokpo', $config['entities']['poke_count']);
         $count = $state['state'] ?? null;
 
         if ($count === null) {