minor code change.