a litter bit change to the script