Fix branch test case