Fix bug with removal animation