Fix branch statement