fix dependency injection