A little code change