Update host_exports.rs

Author: MoonBoi9001

runtime/wasm/src/host_exports.rs

@@ -1258,21 +1258,16 @@ impl HostExports {
         state: &mut BlockState,
     ) -> Result<(), HostExportError> {
         let entity_type = key.entity_type.clone();
-
+        
         if let Some(entity) = state.entity_cache.get(key, GetScope::Store)? {
             state
                 .metrics
                 .track_storage_size_change(&entity_type, &entity, true);
             state.metrics.track_entity_count_change(&entity_type, -1);
         }
 
-        state
-            .entity_cache
-            .modifications
-            .push(EntityModification::remove(
-                key.clone(),
-                state.entity_cache.block_number,
-            ));
+        // Use EntityCache's built-in removal handling
+        state.entity_cache.remove(key.clone())?;
         Ok(())
     }
 }