Corrected a sneaky bug