Update dependency injection