Fixed issue when using ngAnimate and ngClass