fix a small issue