Fix challenge bug