diff --git a/core/datastore-proto/build.gradle.kts b/core/datastore-proto/build.gradle.kts
index be7a16d46..67e9bed02 100644
--- a/core/datastore-proto/build.gradle.kts
+++ b/core/datastore-proto/build.gradle.kts
@@ -15,23 +15,19 @@
  */
 
 plugins {
-    alias(libs.plugins.nowinandroid.android.library)
+    alias(libs.plugins.nowinandroid.jvm.library)
     alias(libs.plugins.protobuf)
 }
 
-android {
-    namespace = "com.google.samples.apps.nowinandroid.core.datastore.proto"
-}
-
 // Setup protobuf configuration, generating lite Java and Kotlin classes
 protobuf {
     protoc {
         artifact = libs.protobuf.protoc.get().toString()
     }
     generateProtoTasks {
-        all().forEach { task ->
-            task.builtins {
-                register("java") {
+        all().configureEach {
+            builtins {
+                named("java") {
                     option("lite")
                 }
                 register("kotlin") {