Update AfterTest.scala