Fix a tweak I made to check out control flow