Update unit tests to use the new injection