Fixed a minor issue with a break / return sub.