fixed small minor bug