add DI wrapper