fix bug in new Test.scala file