Minor code flow fix.