Add injection to make things testable