diff --git a/Project.toml b/Project.toml
index 21434451..6cf039c2 100644
--- a/Project.toml
+++ b/Project.toml
@@ -35,7 +35,7 @@ SpecialFunctions = "1, 2"
 StaticArrays = "1"
 StaticUnivariatePolynomials = "0.6"
 StatsBase = "0.33"
-ValueShapes = "0.9"
+ValueShapes = "0.9, 0.10"
 julia = "1.6"
 
 [extras]
diff --git a/src/BAT.jl b/src/BAT.jl
index 3aa5ce57..8c4f13ae 100644
--- a/src/BAT.jl
+++ b/src/BAT.jl
@@ -50,6 +50,6 @@ function bat_fit(UvD::Type{<:UvSpectrumDensity}, h::Histogram{<:Any, 1};
                 )
             ).result
         )
-    return UvD(mode(bat_samples)[1]), bat_samples
+    return UvD(mode(bat_samples)), bat_samples
 end