Add proper syntax for dependency injection