Fix bug where second collapse is not animated