Corrected a minor bug