small fix to script