diff --git a/features/biter_corpse_remover.lua b/features/biter_corpse_remover.lua index 52cdf460..21bc06e6 100644 --- a/features/biter_corpse_remover.lua +++ b/features/biter_corpse_remover.lua @@ -17,19 +17,9 @@ Global.register(corpse_queue, function(tbl) corpse_queue = tbl end) -local enemy_units = { - ['small-biter'] = true, - ['medium-biter'] = true, - ['big-biter'] = true, - ['behemoth-biter'] = true, - ['small-spitter'] = true, - ['medium-spitter'] = true, - ['big-spitter'] = true, - ['behemoth-spitter'] = true -} - local function entity_died(event) - if not event.unit_number or not enemy_units[event.prototype.name] then + local prototype_type = event.prototype.type + if prototype_type ~= 'unit' and prototype_type ~= 'turret' then return end