diff --git a/plugin/src/sbt-test/sbt-jmh/custom-runner/build.sbt b/plugin/src/sbt-test/sbt-jmh/custom-runner/build.sbt
index 499ac2c..1210cb9 100644
--- a/plugin/src/sbt-test/sbt-jmh/custom-runner/build.sbt
+++ b/plugin/src/sbt-test/sbt-jmh/custom-runner/build.sbt
@@ -1,2 +1,2 @@
 enablePlugins(JmhPlugin)
-mainClass in (Jmh, run) := Some("com.example.CustomRunnerApp")
+Jmh / run / mainClass := Some("com.example.CustomRunnerApp")
diff --git a/plugin/src/sbt-test/sbt-jmh/custom-runner/test b/plugin/src/sbt-test/sbt-jmh/custom-runner/test
index 19445fe..af027f2 100644
--- a/plugin/src/sbt-test/sbt-jmh/custom-runner/test
+++ b/plugin/src/sbt-test/sbt-jmh/custom-runner/test
@@ -1,5 +1,5 @@
 # compile jmh sources all the way to ready to run bench class files
-> jmh:run -i 1 -wi 1 -f 1 .*Hello.*
+> Jmh/run -i 1 -wi 1 -f 1 .*Hello.*
 
 # check if the custom file was generated
 $ exists custom.out
diff --git a/plugin/src/sbt-test/sbt-jmh/generator-error/test b/plugin/src/sbt-test/sbt-jmh/generator-error/test
index d6d58a9..1482ab1 100644
--- a/plugin/src/sbt-test/sbt-jmh/generator-error/test
+++ b/plugin/src/sbt-test/sbt-jmh/generator-error/test
@@ -1,3 +1,3 @@
 # Expect a failure
--> show jmh:managedSources
+-> show Jmh/managedSources
 
diff --git a/plugin/src/sbt-test/sbt-jmh/jmh-asm/test b/plugin/src/sbt-test/sbt-jmh/jmh-asm/test
index 1d52ea1..02af678 100644
--- a/plugin/src/sbt-test/sbt-jmh/jmh-asm/test
+++ b/plugin/src/sbt-test/sbt-jmh/jmh-asm/test
@@ -1,5 +1,5 @@
 # compile jmh sources all the way to ready to run bench class files
-> jmh:run -i 1 -wi 1 -f 1 -rf csv .*Hello.*
+> Jmh/run -i 1 -wi 1 -f 1 -rf csv .*Hello.*
 
 # check if the csv file was generated
 $ exists jmh-result.csv
diff --git a/plugin/src/sbt-test/sbt-jmh/run/test b/plugin/src/sbt-test/sbt-jmh/run/test
index 1d52ea1..02af678 100644
--- a/plugin/src/sbt-test/sbt-jmh/run/test
+++ b/plugin/src/sbt-test/sbt-jmh/run/test
@@ -1,5 +1,5 @@
 # compile jmh sources all the way to ready to run bench class files
-> jmh:run -i 1 -wi 1 -f 1 -rf csv .*Hello.*
+> Jmh/run -i 1 -wi 1 -f 1 -rf csv .*Hello.*
 
 # check if the csv file was generated
 $ exists jmh-result.csv
diff --git a/plugin/src/sbt-test/sbt-jmh/runMain/test b/plugin/src/sbt-test/sbt-jmh/runMain/test
index 85e6073..a33dd96 100644
--- a/plugin/src/sbt-test/sbt-jmh/runMain/test
+++ b/plugin/src/sbt-test/sbt-jmh/runMain/test
@@ -1,5 +1,5 @@
 # use the given class as runner
-> jmh:runMain com.example.CustomRunnerApp -i 1 -wi 1 -f 1 .*Hello.*
+> Jmh/runMain com.example.CustomRunnerApp -i 1 -wi 1 -f 1 .*Hello.*
 
 # check if the custom file was generated
 $ exists custom.out
\ No newline at end of file