minor tweak to script