Minor fix injection fail in unit test