corrected a minor bug