Corrected a silly bug