Improve branch test