diff --git a/src/regressor.jl b/src/regressor.jl
index 18aaef22..9960c90b 100644
--- a/src/regressor.jl
+++ b/src/regressor.jl
@@ -68,5 +68,5 @@ end
 
 MLJModelInterface.metadata_model(MultitargetNeuralNetworkRegressor,
     input=Union{AbstractMatrix{Continuous},Table(Continuous)},
-    target=Table(Continuous),
+    target=Union{AbstractMatrix{Continuous}, Table(Continuous)},
     path="MLJFlux.MultitargetNeuralNetworkRegressor")