Fix ng-class