Correction import to scalatest