Added a missing dependency injection.