small issue fix