Update UnitTest.scala