Replace entity_get_controller() calls