Fixed branch and control_unit state bug