fix challenge unit tests