Adding missing branch tests