Update to minor change in directive API