diff --git a/build.sbt b/build.sbt
index c83f3ae..1c5039c 100644
--- a/build.sbt
+++ b/build.sbt
@@ -120,6 +120,7 @@ lazy val profilingSbtPlugin = project
   .settings(
     name := "sbt-scalac-profiling",
     scalaVersion := bin212.head,
+    sbtPlugin := true,
     scriptedLaunchOpts ++= Seq(
       "-Xmx2048M",
       "-Xms1024M",
@@ -128,7 +129,7 @@ lazy val profilingSbtPlugin = project
     ),
     scriptedBufferLog := false
   )
-  .enablePlugins(SbtPlugin)
+  .enablePlugins(ScriptedPlugin)
 
 // Source dependencies are specified in `project/BuildPlugin.scala`
 lazy val integrations = project
diff --git a/sbt-plugin/src/sbt-test/compiler-profiling/warmup-60-seconds-compile/test b/sbt-plugin/src/sbt-test/compiler-profiling/warmup-60-seconds-compile/test
index 79e8964..f2caeb4 100644
--- a/sbt-plugin/src/sbt-test/compiler-profiling/warmup-60-seconds-compile/test
+++ b/sbt-plugin/src/sbt-test/compiler-profiling/warmup-60-seconds-compile/test
@@ -1,3 +1,3 @@
 > profilingWarmupCompiler
-> set checkCompilerIsWarmedUp in Compile := true
+> set Compile / checkCompilerIsWarmedUp := true
 > profilingWarmupCompiler
diff --git a/sbt-plugin/src/sbt-test/compiler-profiling/warmup-60-seconds-test/test b/sbt-plugin/src/sbt-test/compiler-profiling/warmup-60-seconds-test/test
index 9afdcb8..2ba8d02 100644
--- a/sbt-plugin/src/sbt-test/compiler-profiling/warmup-60-seconds-test/test
+++ b/sbt-plugin/src/sbt-test/compiler-profiling/warmup-60-seconds-test/test
@@ -1,3 +1,3 @@
 > test:profilingWarmupCompiler
-> set checkCompilerIsWarmedUp in Test := true
+> set Test / checkCompilerIsWarmedUp := true
 > test:profilingWarmupCompiler