Fixes dependency injection in new directive