small fix in script