Minor Change

Just a minor code change