minor fix in script