fix small issue