diff --git a/examples/src/main/scala/org/apache/spark/examples/SparkPageRank.scala b/examples/src/main/scala/org/apache/spark/examples/SparkPageRank.scala
index a9b1e6315c5ce738cf837449e5259427b5564a0f..1c1b61d78be753c726719fa7e17027a92be66c90 100644
--- a/examples/src/main/scala/org/apache/spark/examples/SparkPageRank.scala
+++ b/examples/src/main/scala/org/apache/spark/examples/SparkPageRank.scala
@@ -18,7 +18,7 @@
 // scalastyle:off println
 package org.apache.spark.examples
 
-import org.apache.spark.memory
+import org.apache.spark.memory.MemoryMode
 import org.apache.spark.sql.SparkSession
 
 /**