Correct small issue