fix injection unit tests