Fix control flow issue.