tiny code change