Fixing issues with the branch instruction