modify Test.scala