refactor($animate): use $animate.$$setClassImmediately to save code

Please provide a fix that does this while solving the previously outlined issue. Otherwise this is going to be merged in.