diff --git a/gradle/libs.versions.toml b/gradle/libs.versions.toml
index c0732e7..66df7b4 100644
--- a/gradle/libs.versions.toml
+++ b/gradle/libs.versions.toml
@@ -19,7 +19,7 @@ kgp = { module = "org.jetbrains.kotlin.jvm:org.jetbrains.kotlin.jvm.gradle.plugi
 coroutines = "org.jetbrains.kotlinx:kotlinx-coroutines-core:1.10.2"
 gratatouille-runtime = { module = "com.gradleup.gratatouille:gratatouille-runtime", version.ref = "gratatouille-runtime" }
 gratatouille-tasks-runtime = { module = "com.gradleup.gratatouille:gratatouille-tasks-runtime", version.ref = "gratatouille-runtime" }
-nmcp-tasks = {module = "com.gradleup.nmcp:nmcp-tasks", version.ref = "nmcp"}
+nmcp-tasks = { module = "com.gradleup.nmcp:nmcp-tasks", version.ref = "nmcp" }
 
 [plugins]
 kgp = { id = "org.jetbrains.kotlin.jvm", version.ref = "kgp" }
diff --git a/scripts/update-repo.main.kts b/scripts/update-repo.main.kts
index ab5a8dd..c96a4cb 100755
--- a/scripts/update-repo.main.kts
+++ b/scripts/update-repo.main.kts
@@ -3,20 +3,18 @@
 @file:Repository("https://repo.maven.apache.org/maven2/")
 @file:Repository("https://dl.google.com/android/maven2/")
 @file:Repository("https://storage.googleapis.com/gradleup/m2")
-@file:Repository("https://jitpack.io")
 //@file:Repository("file://~/.m2/repository")
-@file:DependsOn("com.gradleup.librarian:librarian-cli:0.2.2-SNAPSHOT-f89d61cb6d4bd03bb9f48d1c8220262c0d02094c")
+@file:DependsOn("com.gradleup.librarian:librarian-cli:0.2.2-SNAPSHOT-b89fe292839b757bf152e8369a88991052d32d0b")
 
-import com.gradleup.librarian.cli.updateRepo
+import com.gradleup.librarian.repo.updateRepo
 
 updateRepo(
-    args,
     setVersion = {
         file("nmcp/testProjec/duplicate-name/build.gradle.kts") {
             replacePluginVersion("com.gradleup.nmcp.aggregation")
         }
     },
-    setDocsVersion = {
+    setVersionInDocs = {
         file("docs/src/content/docs/index.mdx") {
             replacePluginVersion("com.gradleup.nmcp.settings")
             replacePluginVersion("com.gradleup.nmcp.aggregation")