A small code change