diff --git a/docker-compose.yml b/docker-compose.yml
index 1be391a..b59b5e4 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -44,7 +44,7 @@
     environment:
       - "SPARK_MASTER=spark://spark-master:7077"
       - "SPARK_MASTER_WEBUI_PORT=8080"
-      - "PYSPARK_SUBMIT_ARGS=--master spark://spark-master:7077 --packages org.apache.spark:spark-sql-kafka-0-10_2.11:2.4.3 pyspark-shell"
+      - "PYSPARK_SUBMIT_ARGS=--master spark://spark-master:7077 --packages org.apache.spark:spark-sql-kafka-0-10_2.11:2.4.3,graphframes:graphframes:0.8.1-spark2.4-s_2.11 pyspark-shell"
     working_dir: /home/pyspark/work
     volumes:
       - ${HOME}/tmp/sparkdata:/mnt/sparkdata