diff --git a/.github/ISSUE_TEMPLATE/bug-report.yml b/.github/ISSUE_TEMPLATE/bug-report.yml
index 60e28f8..7f28c31 100644
--- a/.github/ISSUE_TEMPLATE/bug-report.yml
+++ b/.github/ISSUE_TEMPLATE/bug-report.yml
@@ -34,7 +34,7 @@ body:
             description: |
                 If your Minecraft version isn't listed here, it means that it's no longer supported. In that case, don't create an issue.
             options:
-                - Minecraft 1.20.4
+                - Minecraft 1.21
         validations:
             required: true
     -   type: input
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index 53ca679..a157d3d 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -8,7 +8,7 @@ on:
     types: [ opened, synchronize, reopened ]
 jobs:
   build:
-    uses: refinedmods/refinedarchitect/.github/workflows/build.yml@v0.15.1
+    uses: refinedmods/refinedarchitect/.github/workflows/build.yml@v0.16.4
     with:
       mutation-testing: false
     secrets: inherit
diff --git a/.github/workflows/draft-release.yml b/.github/workflows/draft-release.yml
index f1a44f0..2300f88 100644
--- a/.github/workflows/draft-release.yml
+++ b/.github/workflows/draft-release.yml
@@ -17,7 +17,7 @@ on:
         type: string
 jobs:
   draft:
-    uses: refinedmods/refinedarchitect/.github/workflows/draft-release.yml@v0.15.1
+    uses: refinedmods/refinedarchitect/.github/workflows/draft-release.yml@v0.16.4
     with:
       release-type: ${{ inputs.release-type }}
       version-number-override: ${{ inputs.version-number-override }}
diff --git a/.github/workflows/issue-for-unsupported-version.yml b/.github/workflows/issue-for-unsupported-version.yml
index 68f5d19..7444dec 100644
--- a/.github/workflows/issue-for-unsupported-version.yml
+++ b/.github/workflows/issue-for-unsupported-version.yml
@@ -4,4 +4,4 @@ on:
     types: [ labeled, unlabeled, reopened ]
 jobs:
   unsupported-labeler:
-    uses: refinedmods/refinedarchitect/.github/workflows/issue-for-unsupported-version.yml@v0.15.1
\ No newline at end of file
+    uses: refinedmods/refinedarchitect/.github/workflows/issue-for-unsupported-version.yml@v0.16.4
\ No newline at end of file
diff --git a/.github/workflows/publish-release.yml b/.github/workflows/publish-release.yml
index d5a4fcf..b2b9bdf 100644
--- a/.github/workflows/publish-release.yml
+++ b/.github/workflows/publish-release.yml
@@ -7,7 +7,7 @@ on:
       - closed
 jobs:
   publish-release:
-    uses: refinedmods/refinedarchitect/.github/workflows/publish-release.yml@v0.15.1
+    uses: refinedmods/refinedarchitect/.github/workflows/publish-release.yml@v0.16.4
     secrets: inherit
     with:
       project-name: 'Refined Storage - JEI Integration'
diff --git a/.github/workflows/resolved-issue-locking.yml b/.github/workflows/resolved-issue-locking.yml
index f40e8ed..ee10c5c 100644
--- a/.github/workflows/resolved-issue-locking.yml
+++ b/.github/workflows/resolved-issue-locking.yml
@@ -4,4 +4,4 @@ on:
     - cron: '0 0 * * *'
 jobs:
   lock:
-    uses: refinedmods/refinedarchitect/.github/workflows/resolved-issue-locking.yml@v0.15.1
\ No newline at end of file
+    uses: refinedmods/refinedarchitect/.github/workflows/resolved-issue-locking.yml@v0.16.4
\ No newline at end of file
diff --git a/.github/workflows/validate-branch-name.yml b/.github/workflows/validate-branch-name.yml
index 0e3e2e1..1614152 100644
--- a/.github/workflows/validate-branch-name.yml
+++ b/.github/workflows/validate-branch-name.yml
@@ -2,4 +2,4 @@ name: Validate branch name
 on: [ pull_request ]
 jobs:
   validate-branch-name:
-    uses: refinedmods/refinedarchitect/.github/workflows/validate-branch-name.yml@v0.15.1
\ No newline at end of file
+    uses: refinedmods/refinedarchitect/.github/workflows/validate-branch-name.yml@v0.16.4
\ No newline at end of file
diff --git a/.github/workflows/validate-changelog.yml b/.github/workflows/validate-changelog.yml
index e16cf7e..d2a2d80 100644
--- a/.github/workflows/validate-changelog.yml
+++ b/.github/workflows/validate-changelog.yml
@@ -4,4 +4,4 @@ on:
     types: [ opened, synchronize, reopened, ready_for_review, labeled, unlabeled ]
 jobs:
   validate-changelog:
-    uses: refinedmods/refinedarchitect/.github/workflows/validate-changelog.yml@v0.15.1
\ No newline at end of file
+    uses: refinedmods/refinedarchitect/.github/workflows/validate-changelog.yml@v0.16.4
\ No newline at end of file
diff --git a/.github/workflows/validate-commit-messages.yml b/.github/workflows/validate-commit-messages.yml
index dbef381..449efa5 100644
--- a/.github/workflows/validate-commit-messages.yml
+++ b/.github/workflows/validate-commit-messages.yml
@@ -2,4 +2,4 @@ name: Validate commit messages
 on: [ pull_request ]
 jobs:
   validate-commit-messages:
-    uses: refinedmods/refinedarchitect/.github/workflows/validate-commit-messages.yml@v0.15.1
\ No newline at end of file
+    uses: refinedmods/refinedarchitect/.github/workflows/validate-commit-messages.yml@v0.16.4
\ No newline at end of file
diff --git a/.gitignore b/.gitignore
index 924b02b..8c2b9ee 100644
--- a/.gitignore
+++ b/.gitignore
@@ -15,3 +15,4 @@ out/
 /bin/
 logs/
 .cache/
+run/
diff --git a/CHANGELOG.md b/CHANGELOG.md
index 0746445..b9e8864 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -7,6 +7,8 @@ to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
 ## [Unreleased]
 
+## [0.3.1] - 2024-07-10
+
 ## [0.3.0] - 2024-07-06
 
 ### Fixed
@@ -42,7 +44,9 @@ to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 -   Recipe transfer integration for the Crafting Grid.
 -   Grid / JEI search box synchronization options.
 
-[Unreleased]: https://github.com/refinedmods/refinedstorage-jei-integration/compare/v0.3.0...HEAD
+[Unreleased]: https://github.com/refinedmods/refinedstorage-jei-integration/compare/v0.3.1...HEAD
+
+[0.3.1]: https://github.com/refinedmods/refinedstorage-jei-integration/compare/v0.3.0...v0.3.1
 
 [0.3.0]: https://github.com/refinedmods/refinedstorage-jei-integration/compare/v0.2.2...v0.3.0
 
diff --git a/build.gradle b/build.gradle
deleted file mode 100644
index 7f2727d..0000000
--- a/build.gradle
+++ /dev/null
@@ -1,7 +0,0 @@
-apply from: "https://raw.githubusercontent.com/refinedmods/refinedarchitect/v${gradle.ext.refinedArchitectVersion}/helper.gradle"
-
-subprojects {
-    group = 'com.refinedmods.refinedstorage'
-}
-
-enableSonarQube("refinedmods_refinedstorage-jei-integration")
\ No newline at end of file
diff --git a/build.gradle.kts b/build.gradle.kts
new file mode 100644
index 0000000..28b6b8c
--- /dev/null
+++ b/build.gradle.kts
@@ -0,0 +1,12 @@
+plugins {
+    id("refinedarchitect.root")
+    id("refinedarchitect.base")
+}
+
+refinedarchitect {
+    sonarQube("refinedmods_refinedstorage-jei-integration", "refinedmods")
+}
+
+subprojects {
+    group = "com.refinedmods.refinedstorage"
+}
diff --git a/gradle.properties b/gradle.properties
index 8acfd86..40da4ed 100644
--- a/gradle.properties
+++ b/gradle.properties
@@ -1,4 +1,5 @@
-refinedstorageVersion=2.0.0-milestone.4.1
-jeiVersion=19.1.0.17
+refinedstorageVersion=2.0.0-milestone.4.4
+jeiVersion=19.4.0.28
+minecraftVersion=1.21
 # Gradle
 org.gradle.jvmargs=-Xmx1G
diff --git a/refinedstorage-jei-integration-common/build.gradle b/refinedstorage-jei-integration-common/build.gradle
deleted file mode 100644
index a731684..0000000
--- a/refinedstorage-jei-integration-common/build.gradle
+++ /dev/null
@@ -1,29 +0,0 @@
-plugins {
-    id 'java-library'
-}
-
-archivesBaseName = 'refinedstorage-jei-integration-common'
-
-repositories {
-    maven {
-        url = uri("https://maven.pkg.github.com/refinedmods/refinedstorage2")
-        credentials {
-            username = "anything"
-            password = "\u0067hp_oGjcDFCn8jeTzIj4Ke9pLoEVtpnZMP4VQgaX"
-        }
-    }
-    maven {
-        name = 'JEI'
-        url = "https://maven.blamejared.com/"
-    }
-}
-
-commonProject()
-
-dependencies {
-    api "com.refinedmods.refinedstorage:refinedstorage-platform-common:${refinedstorageVersion}"
-    api "mezz.jei:jei-${minecraftVersion}-common-api:${jeiVersion}"
-    api "mezz.jei:jei-${minecraftVersion}-common:${jeiVersion}"
-}
-
-enablePublishing()
\ No newline at end of file
diff --git a/refinedstorage-jei-integration-common/build.gradle.kts b/refinedstorage-jei-integration-common/build.gradle.kts
new file mode 100644
index 0000000..724338d
--- /dev/null
+++ b/refinedstorage-jei-integration-common/build.gradle.kts
@@ -0,0 +1,38 @@
+plugins {
+    id("refinedarchitect.common")
+}
+
+repositories {
+    maven {
+        url = uri("https://maven.pkg.github.com/refinedmods/refinedstorage2")
+        credentials {
+            username = "anything"
+            password = "\u0067hp_oGjcDFCn8jeTzIj4Ke9pLoEVtpnZMP4VQgaX"
+        }
+    }
+    maven {
+        name = "JEI"
+        url = uri("https://maven.blamejared.com/")
+    }
+}
+
+refinedarchitect {
+    common()
+    publishing {
+        maven = true
+    }
+}
+
+base {
+    archivesName.set("refinedstorage-jei-integration-common")
+}
+
+val refinedstorageVersion: String by project
+val jeiVersion: String by project
+val minecraftVersion: String by project
+
+dependencies {
+    api("com.refinedmods.refinedstorage:refinedstorage-platform-common:${refinedstorageVersion}")
+    api("mezz.jei:jei-${minecraftVersion}-common-api:${jeiVersion}")
+    api("mezz.jei:jei-${minecraftVersion}-common:${jeiVersion}")
+}
diff --git a/refinedstorage-jei-integration-fabric/.gitignore b/refinedstorage-jei-integration-fabric/.gitignore
deleted file mode 100644
index 737e26b..0000000
--- a/refinedstorage-jei-integration-fabric/.gitignore
+++ /dev/null
@@ -1 +0,0 @@
-run/
\ No newline at end of file
diff --git a/refinedstorage-jei-integration-fabric/build.gradle b/refinedstorage-jei-integration-fabric/build.gradle
deleted file mode 100644
index a1ce81c..0000000
--- a/refinedstorage-jei-integration-fabric/build.gradle
+++ /dev/null
@@ -1,38 +0,0 @@
-archivesBaseName = 'refinedstorage-jei-integration-fabric'
-
-fabricProject("refinedstorage_jei_integration", project(":refinedstorage-jei-integration-common"))
-
-repositories {
-    maven {
-        url = uri("https://maven.pkg.github.com/refinedmods/refinedstorage2")
-        credentials {
-            username = "anything"
-            password = "\u0067hp_oGjcDFCn8jeTzIj4Ke9pLoEVtpnZMP4VQgaX"
-        }
-    }
-    maven {
-        name = 'ModMenu'
-        url 'https://maven.terraformersmc.com/'
-    }
-    maven {
-        name = 'Cloth Config'
-        url 'https://maven.shedaniel.me/'
-    }
-    maven {
-        name = 'NoIndium'
-        url = 'https://maven.cafeteria.dev/releases/'
-    }
-    maven {
-        name = 'JEI'
-        url = "https://maven.blamejared.com/"
-    }
-}
-
-dependencies {
-    modApi "com.refinedmods.refinedstorage:refinedstorage-platform-fabric:${refinedstorageVersion}"
-    modRuntimeOnly "mezz.jei:jei-${minecraftVersion}-fabric:${jeiVersion}"
-    modCompileOnlyApi "mezz.jei:jei-${minecraftVersion}-common-api:${jeiVersion}"
-    modCompileOnlyApi "mezz.jei:jei-${minecraftVersion}-fabric-api:${jeiVersion}"
-}
-
-enablePublishing()
\ No newline at end of file
diff --git a/refinedstorage-jei-integration-fabric/build.gradle.kts b/refinedstorage-jei-integration-fabric/build.gradle.kts
new file mode 100644
index 0000000..7cc623c
--- /dev/null
+++ b/refinedstorage-jei-integration-fabric/build.gradle.kts
@@ -0,0 +1,50 @@
+plugins {
+    id("refinedarchitect.fabric")
+}
+
+repositories {
+    maven {
+        url = uri("https://maven.pkg.github.com/refinedmods/refinedstorage2")
+        credentials {
+            username = "anything"
+            password = "\u0067hp_oGjcDFCn8jeTzIj4Ke9pLoEVtpnZMP4VQgaX"
+        }
+    }
+    maven {
+        name = "ModMenu"
+        url = uri("https://maven.terraformersmc.com/")
+    }
+    maven {
+        name = "Cloth Config"
+        url = uri("https://maven.shedaniel.me/")
+    }
+    maven {
+        name = "JEI"
+        url = uri("https://maven.blamejared.com/")
+    }
+}
+
+refinedarchitect {
+    modId = "refinedstorage_jei_integration"
+    fabric()
+    compileWithProject(project(":refinedstorage-jei-integration-common"))
+    publishing {
+        maven = true
+    }
+}
+
+base {
+    archivesName.set("refinedstorage-jei-integration-fabric")
+}
+
+val refinedstorageVersion: String by project
+val jeiVersion: String by project
+val minecraftVersion: String by project
+
+dependencies {
+    modApi("com.refinedmods.refinedstorage:refinedstorage-platform-fabric:${refinedstorageVersion}")
+    modRuntimeOnly("mezz.jei:jei-${minecraftVersion}-fabric:${jeiVersion}")
+    modCompileOnlyApi("mezz.jei:jei-${minecraftVersion}-common-api:${jeiVersion}")
+    modCompileOnlyApi("mezz.jei:jei-${minecraftVersion}-common:${jeiVersion}")
+    modCompileOnlyApi("mezz.jei:jei-${minecraftVersion}-fabric-api:${jeiVersion}")
+}
diff --git a/refinedstorage-jei-integration-neoforge/.gitignore b/refinedstorage-jei-integration-neoforge/.gitignore
deleted file mode 100644
index 737e26b..0000000
--- a/refinedstorage-jei-integration-neoforge/.gitignore
+++ /dev/null
@@ -1 +0,0 @@
-run/
\ No newline at end of file
diff --git a/refinedstorage-jei-integration-neoforge/build.gradle b/refinedstorage-jei-integration-neoforge/build.gradle
deleted file mode 100644
index aa7b197..0000000
--- a/refinedstorage-jei-integration-neoforge/build.gradle
+++ /dev/null
@@ -1,30 +0,0 @@
-plugins {
-    id 'java-library'
-}
-
-archivesBaseName = 'refinedstorage-jei-integration-neoforge'
-
-neoForgeProject("refinedstorage_jei_integration", project(":refinedstorage-jei-integration-common"))
-
-repositories {
-    maven {
-        url = uri("https://maven.pkg.github.com/refinedmods/refinedstorage2")
-        credentials {
-            username = "anything"
-            password = "\u0067hp_oGjcDFCn8jeTzIj4Ke9pLoEVtpnZMP4VQgaX"
-        }
-    }
-    maven {
-        name = 'JEI'
-        url = "https://maven.blamejared.com/"
-    }
-}
-
-dependencies {
-    api "com.refinedmods.refinedstorage:refinedstorage-platform-neoforge:${refinedstorageVersion}"
-    runtimeOnly "mezz.jei:jei-${minecraftVersion}-neoforge:${jeiVersion}"
-    compileOnlyApi "mezz.jei:jei-${minecraftVersion}-common-api:${jeiVersion}"
-    compileOnlyApi "mezz.jei:jei-${minecraftVersion}-neoforge-api:${jeiVersion}"
-}
-
-enablePublishing()
\ No newline at end of file
diff --git a/refinedstorage-jei-integration-neoforge/build.gradle.kts b/refinedstorage-jei-integration-neoforge/build.gradle.kts
new file mode 100644
index 0000000..f993e5a
--- /dev/null
+++ b/refinedstorage-jei-integration-neoforge/build.gradle.kts
@@ -0,0 +1,41 @@
+plugins {
+    id("refinedarchitect.neoforge")
+}
+
+repositories {
+    maven {
+        url = uri("https://maven.pkg.github.com/refinedmods/refinedstorage2")
+        credentials {
+            username = "anything"
+            password = "\u0067hp_oGjcDFCn8jeTzIj4Ke9pLoEVtpnZMP4VQgaX"
+        }
+    }
+    maven {
+        name = "JEI"
+        url = uri("https://maven.blamejared.com/")
+    }
+}
+
+refinedarchitect {
+    modId = "refinedstorage_jei_integration"
+    neoForge()
+    compileWithProject(project(":refinedstorage-jei-integration-common"))
+    publishing {
+        maven = true
+    }
+}
+
+base {
+    archivesName.set("refinedstorage-jei-integration-neoforge")
+}
+
+val refinedstorageVersion: String by project
+val jeiVersion: String by project
+val minecraftVersion: String by project
+
+dependencies {
+    api("com.refinedmods.refinedstorage:refinedstorage-platform-neoforge:${refinedstorageVersion}")
+    runtimeOnly("mezz.jei:jei-${minecraftVersion}-neoforge:${jeiVersion}")
+    compileOnlyApi("mezz.jei:jei-${minecraftVersion}-common-api:${jeiVersion}")
+    compileOnlyApi("mezz.jei:jei-${minecraftVersion}-neoforge-api:${jeiVersion}")
+}
diff --git a/settings.gradle b/settings.gradle
deleted file mode 100644
index f9e50b7..0000000
--- a/settings.gradle
+++ /dev/null
@@ -1,18 +0,0 @@
-gradle.ext.refinedArchitectVersion = "0.15.1"
-
-dependencyResolutionManagement {
-    repositories {
-        maven {
-            url = "https://maven.pkg.github.com/refinedmods/refinedarchitect"
-            credentials {
-                username = "anything"
-                password = "\u0067hp_oGjcDFCn8jeTzIj4Ke9pLoEVtpnZMP4VQgaX"
-            }
-        }
-    }
-}
-
-rootProject.name = 'refinedstorage-jei-integration'
-include 'refinedstorage-jei-integration-common'
-include 'refinedstorage-jei-integration-neoforge'
-include 'refinedstorage-jei-integration-fabric'
diff --git a/settings.gradle.kts b/settings.gradle.kts
new file mode 100644
index 0000000..31b3be3
--- /dev/null
+++ b/settings.gradle.kts
@@ -0,0 +1,33 @@
+pluginManagement {
+    repositories {
+        mavenCentral()
+        gradlePluginPortal()
+        maven {
+            url = uri("https://maven.pkg.github.com/refinedmods/refinedarchitect")
+            credentials {
+                username = "anything"
+                password = "\u0067hp_oGjcDFCn8jeTzIj4Ke9pLoEVtpnZMP4VQgaX"
+            }
+        }
+        maven {
+            name = "Fabric"
+            url = uri("https://maven.fabricmc.net/")
+        }
+        maven {
+            name = "NeoForge"
+            url = uri("https://maven.neoforged.net/releases")
+        }
+    }
+    plugins {
+        id("refinedarchitect.root").version("0.16.4")
+        id("refinedarchitect.base").version("0.16.4")
+        id("refinedarchitect.common").version("0.16.4")
+        id("refinedarchitect.neoforge").version("0.16.4")
+        id("refinedarchitect.fabric").version("0.16.4")
+    }
+}
+
+rootProject.name = "refinedstorage-jei-integration"
+include("refinedstorage-jei-integration-common")
+include("refinedstorage-jei-integration-neoforge")
+include("refinedstorage-jei-integration-fabric")