Update tests for using element.removeClass instead of $animate