#18 Additional DI fixes