Fix entity enable/disable