Slight tweak to decision loop