Fix ngClass animation bugs with enter, move and leave