diff --git a/Project.toml b/Project.toml
index bcddeca..433858d 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,7 +1,7 @@
 name = "MLJFlux"
 uuid = "094fc8d1-fd35-5302-93ea-dabda2abf845"
 authors = ["Anthony D. Blaom <anthony.blaom@gmail.com>", "Ayush Shridhar <ayush.shridhar1999@gmail.com>"]
-version = "0.6.1"
+version = "0.6.2"
 
 [deps]
 CategoricalArrays = "324d7699-5711-5eae-9e2f-1d82baa6b597"
@@ -21,7 +21,7 @@ CategoricalArrays = "0.10"
 ColorTypes = "0.10.3, 0.11"
 ComputationalResources = "0.3.2"
 Flux = "0.14"
-MLJModelInterface = "1.1.1"
+MLJModelInterface = "1.11"
 Metalhead = "0.9.3"
 Optimisers = "0.3.2"
 ProgressMeter = "1.7.1"
diff --git a/src/mlj_embedder_interface.jl b/src/mlj_embedder_interface.jl
index b37a90d..6be5df6 100644
--- a/src/mlj_embedder_interface.jl
+++ b/src/mlj_embedder_interface.jl
@@ -36,6 +36,7 @@ MMI.metadata_pkg(
     package_uuid = "23777cdb-d90c-4eb0-a694-7c2b83d5c1d6",
     package_url = "https://github.com/JuliaAI/MLJTransforms.jl",
     is_pure_julia = true,
+    is_wrapper = true
 )
 
 MMI.metadata_model(
@@ -46,7 +47,6 @@ MMI.metadata_model(
 MMI.target_in_fit(::Type{<:EntityEmbedder}) = true
 
 # 9. Forwarding traits
-MMI.is_wrapper(::Type{<:EntityEmbedder}) =true
 MMI.supports_training_losses(::Type{<:EntityEmbedder}) = true