Update deprecated scalatest import