fixed small issue

fixed small issue