Just a minor script change I made while working.