minor code change to conditional