Fixing bug with script arguments