Reference to ngAnimate now inside directive instead of main app