Bugfix: Correcting conditional for branch code check