diff --git a/build.sbt b/build.sbt
index 560eb77..9010379 100644
--- a/build.sbt
+++ b/build.sbt
@@ -1,6 +1,6 @@
 name := "template-scala-parallel-classification"
-
+scalaVersion := "2.11.8"
 libraryDependencies ++= Seq(
   "org.apache.predictionio" %% "apache-predictionio-core" % "0.11.0-incubating" % "provided",
-  "org.apache.spark"        %% "spark-core"               % "1.3.1" % "provided",
-  "org.apache.spark"        %% "spark-mllib"              % "1.3.1" % "provided")
+  "org.apache.spark"        %% "spark-core"               % "2.1.1" % "provided",
+  "org.apache.spark"        %% "spark-mllib"              % "2.1.1" % "provided")