Minor saftey ajust.