diff --git a/mllib/src/test/scala/org/apache/spark/mllib/fpm/FPGrowthSuite.scala b/mllib/src/test/scala/org/apache/spark/mllib/fpm/FPGrowthSuite.scala
index aad5b4a3f2afb..e29399ffe71fe 100644
--- a/mllib/src/test/scala/org/apache/spark/mllib/fpm/FPGrowthSuite.scala
+++ b/mllib/src/test/scala/org/apache/spark/mllib/fpm/FPGrowthSuite.scala
@@ -17,9 +17,9 @@
 package org.apache.spark.mllib.fpm
 
 import org.scalatest.FunSuite
-import org.apache.spark.mllib.util.LocalSparkContext
+import org.apache.spark.mllib.util.MLlibTestSparkContext
 
-class FPGrowthSuite  extends FunSuite with LocalSparkContext {
+class FPGrowthSuite  extends FunSuite with MLlibTestSparkContext {
 
   test("test FPGrowth algorithm")
   {