diff --git a/with-spark.def b/with-spark.def
index 512fbcd..16cdf96 100644
--- a/with-spark.def
+++ b/with-spark.def
@@ -10,7 +10,7 @@ From: mobileinsight.sif
         python3-pandas \
         openjdk-8-jre-headless \
         scala
-    pip3 install pyspark==3.5.1
+    pip3 install pyspark[sql]==3.5.1 dill==0.3.8
 
     # Cleanup
     apt-get autoclean -y