modify test.scala