small code change