edited a minor bug