Minor alteration to fix minor error