diff --git a/R/dependencies.R b/R/dependencies.R
index bbfab9f..523d139 100644
--- a/R/dependencies.R
+++ b/R/dependencies.R
@@ -1,5 +1,9 @@
 spark_dependencies <- function(spark_version, scala_version, ...) {
-  graphframes_version <- if (spark_version >= "2.4.0") {
+  graphframes_version <- if (spark_version >= "3.1.0") {
+    "0.8.2"
+  } else if (spark_version >= "3.0.0") {
+    "0.8.0"
+  } else if (spark_version >= "2.4.0") {
     "0.7.0"
   } else if (spark_version >= "2.2.0") {
     "0.6.0"