diff --git a/.github/workflows/gradle-publish.yml b/.github/workflows/gradle-publish.yml
index 3feef94..4036012 100644
--- a/.github/workflows/gradle-publish.yml
+++ b/.github/workflows/gradle-publish.yml
@@ -17,7 +17,7 @@ jobs:
   build:
 
     runs-on: ubuntu-latest
-    container: wpilib/roborio-cross-ubuntu:2022-18.04
+    container: wpilib/roborio-cross-ubuntu:2024-22.04
     permissions:
       contents: read
       packages: write
@@ -42,7 +42,7 @@ jobs:
     - name: Publish to GitHub Packages
       uses: gradle/gradle-build-action@0d13054264b0bb894ded474f08ebb30921341cee
       with:
-        arguments: publish
+        arguments: -Prelease publish
       env:
         USERNAME: ${{ github.actor }}
         TOKEN: ${{ secrets.GITHUB_TOKEN }}
diff --git a/lib/publish.gradle b/lib/publish.gradle
index 83f15bd..b712d89 100644
--- a/lib/publish.gradle
+++ b/lib/publish.gradle
@@ -57,16 +57,20 @@ task cleanReleaseRepo(type: Delete) {
 
 tasks.matching {it != cleanReleaseRepo}.all {it.dependsOn cleanReleaseRepo}
 
-model {
-    publishing {
-        publications {
-            java(MavenPublication) {
-                artifactId = "${baseArtifactId}-java"
-                groupId artifactGroupId
-                version pubVersion
+task sourceJar(type: Jar) {
+    from sourceSets.main.allJava
+    archiveClassifier = "sources"
+}
 
-                from components.java
-            }
+publishing {
+    publications {
+        maven(MavenPublication) {
+            artifact sourceJar
+            artifactId = "${baseArtifactId}-java"
+            groupId artifactGroupId
+            version = project.hasProperty('release') ? pubVersion : pubVersion + "-SNAPSHOT"
+
+            from components.java
         }
     }
 }