diff --git a/docs/make.jl b/docs/make.jl
index 234e065..cc3684e 100644
--- a/docs/make.jl
+++ b/docs/make.jl
@@ -1,7 +1,7 @@
 using Maxnet
 using Documenter
 
-DocMeta.setdocmeta!(Maxnet, :DocTestSetup, :(using Maxnet); recursive=true)
+DocMeta.setdocmeta!(Maxnet, :DocTestSetup, :(using Maxnet; using MLJBase: Multiclass); recursive=true)
 
 makedocs(;
     modules=[Maxnet],