diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
index 1601cf1..d7d297c 100644
--- a/.github/workflows/release.yml
+++ b/.github/workflows/release.yml
@@ -8,10 +8,10 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v1
-      - name: Set up JDK 1.8
+      - name: Set up JDK 8
         uses: actions/setup-java@v3
         with:
-            java-version: 1.8
+            java-version: 8
             distribution: temurin
       - uses: olafurpg/setup-gpg@v3
       - name: Publish ${{ github.ref }}
diff --git a/build.sbt b/build.sbt
index 10c57a7..149f6f6 100644
--- a/build.sbt
+++ b/build.sbt
@@ -90,7 +90,8 @@ lazy val `sparksql-scalapb` = (projectMatrix in file("sparksql-scalapb"))
     Test / PB.targets := Seq(
       scalapbPlugin(scalapb.value.scalapbVersion) -> (Test / sourceManaged).value
     ),
-    Test / run / fork := true
+    Test / run / fork := true,
+    Test / javaOptions ++= Seq("-Xmx2G"),
   )
   .customRow(
     scalaVersions = Seq(Scala212, Scala213),