diff --git a/core/src/test/scala/spark/RDDSuite.scala b/core/src/test/scala/spark/RDDSuite.scala
index ef74c99246728a2b4857914c26af9ac43cdc997b..5a3a12dfff8e34c2cd7bbad1213e1905e22d129f 100644
--- a/core/src/test/scala/spark/RDDSuite.scala
+++ b/core/src/test/scala/spark/RDDSuite.scala
@@ -174,9 +174,9 @@ class RDDSuite extends FunSuite with BeforeAndAfter {
     val data = sc.parallelize(1 to 10, 10)
     // Note that split number starts from 0, so > 8 means only 10th partition left.
     val prunedRdd = data.pruneSplits(splitNum => splitNum > 8)
-    assert(prunedRdd.splits.size == 1)
+    assert(prunedRdd.splits.size === 1)
     val prunedData = prunedRdd.collect
-    assert(prunedData.size == 1)
-    assert(prunedData(0) == 10)
+    assert(prunedData.size === 1)
+    assert(prunedData(0) === 10)
   }
 }