diff --git a/Project.toml b/Project.toml
index 1d8c19a3..efdaf2a4 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,6 +1,6 @@
 name = "AdvancedHMC"
 uuid = "0bf59076-c3b1-5ca4-86bd-e02cd72cde3d"
-version = "0.6.0"
+version = "0.6.1"
 
 [deps]
 AbstractMCMC = "80f14c24-f653-4e6a-9b94-39d6b0f70001"
diff --git a/test/runtests.jl b/test/runtests.jl
index 382bb871..edc9fa6f 100644
--- a/test/runtests.jl
+++ b/test/runtests.jl
@@ -36,7 +36,7 @@ if GROUP == "All" || GROUP == "AdvancedHMC"
         @warn "Skipping GPU tests because no GPU available."
     end
 
-    @main function runtests(patterns...; dry::Bool = false)
+    Comonicon.@main function runtests(patterns...; dry::Bool = false)
         retest(patterns...; dry = dry, verbose = Inf)
     end
 end