Adding another test to make sure each branch is covered