diff --git a/includes/discovery/entity-state.inc.php b/includes/discovery/entity-state.inc.php
index 2bfaad3e51..5cc38d7bb6 100644
--- a/includes/discovery/entity-state.inc.php
+++ b/includes/discovery/entity-state.inc.php
@@ -56,7 +56,7 @@ if (! empty($entPhysical)) {
 
             if (isset($db_states[$id])) { // update the db
                 $db_state = $db_states[$id];
-                $update = array_diff($state, $db_state);
+                $update = array_diff_assoc($state, $db_state);
 
                 if (! empty($update)) {
                     if (array_key_exists('entStateLastChanged', $update) && is_null($update['entStateLastChanged'])) {
diff --git a/includes/polling/entity-state.inc.php b/includes/polling/entity-state.inc.php
index fc446d3579..61b0e06e37 100644
--- a/includes/polling/entity-state.inc.php
+++ b/includes/polling/entity-state.inc.php
@@ -64,7 +64,7 @@ if (! empty($entityStatesIndexes)) {
                         ->format('Y-m-d H:i:s');
 
                     // check if anything has changed
-                    $update = array_diff(
+                    $update = array_diff_assoc(
                         $new_states,
                         dbFetchRow(
                             'SELECT * FROM entityState WHERE entity_state_id=?',