Update BranchStatementTest.scala