minor cosmetic fix to script