Add missing dependency Injection