diff --git a/jupyter/src/main/kotlin/org/jetbrains/kotlinx/spark/api/jupyter/Integration.kt b/jupyter/src/main/kotlin/org/jetbrains/kotlinx/spark/api/jupyter/Integration.kt
index b35109c8..448751ae 100644
--- a/jupyter/src/main/kotlin/org/jetbrains/kotlinx/spark/api/jupyter/Integration.kt
+++ b/jupyter/src/main/kotlin/org/jetbrains/kotlinx/spark/api/jupyter/Integration.kt
@@ -84,6 +84,7 @@ abstract class Integration(private val notebook: Notebook, private val options:
         "org.jetbrains.kotlin:kotlin-stdlib-jdk8:$kotlinVersion",
         "org.jetbrains.kotlin:kotlin-reflect:$kotlinVersion",
         "org.apache.spark:spark-sql_$scalaCompatVersion:$sparkVersion",
+        "org.apache.spark:spark-yarn_$scalaCompatVersion:$sparkVersion",
         "org.apache.spark:spark-streaming_$scalaCompatVersion:$sparkVersion",
         "org.apache.spark:spark-mllib_$scalaCompatVersion:$sparkVersion",
         "org.apache.spark:spark-sql_$scalaCompatVersion:$sparkVersion",