fix small omission in script