fix challenge problem