Small code change