diff --git a/mllib-local/src/test/scala/org/apache/spark/ml/DummyTestingSuite.scala b/mllib-local/src/test/scala/org/apache/spark/ml/DummyTestingSuite.scala
index 6c76dbfbfa9e0..8fc03aadafb3b 100644
--- a/mllib-local/src/test/scala/org/apache/spark/ml/DummyTestingSuite.scala
+++ b/mllib-local/src/test/scala/org/apache/spark/ml/DummyTestingSuite.scala
@@ -17,10 +17,10 @@
 
 package org.apache.spark.ml
 
-import org.apache.spark.SparkFunSuite
+import org.scalatest.{BeforeAndAfterAll, FunSuite, Outcome}
 
 // This is testing if the new build works. To be removed soon.
-class DummyTestingSuite extends SparkFunSuite {
+class DummyTestingSuite extends FunSuite {
 
   test("This is testing if the new build works.") {
     assert(DummyTesting.add10(15) === 25)