Fix control flow example