That wasn't enough. Had to handle the reverse animation correctly