Correct small mistake