corrected minor bug