fixing challenge bug