diff --git a/Project.toml b/Project.toml
index f12de2c..c368817 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,7 +1,7 @@
 name = "TaijaPlotting"
 uuid = "bd7198b4-c7d6-400c-9bab-9a24614b0240"
 authors = ["Patrick Altmeyer"]
-version = "1.0.4"
+version = "1.0.5"
 
 [deps]
 CategoricalArrays = "324d7699-5711-5eae-9e2f-1d82baa6b597"
diff --git a/src/ConformalPrediction/ConformalPrediction.jl b/src/ConformalPrediction/ConformalPrediction.jl
index aa4f310..d1b3acd 100644
--- a/src/ConformalPrediction/ConformalPrediction.jl
+++ b/src/ConformalPrediction/ConformalPrediction.jl
@@ -129,7 +129,7 @@ function Plots.contourf(
             z = ismissing(p̂) ? 0 : sum(pdf.(p̂, p̂.decoder.classes) .> 0)
         elseif plot_classification_loss
             _target = categorical([target]; levels=levels(y))
-            z = ConformalPrediction.classification_loss(
+            z = ConformalPrediction.ConformalTraining.classification_loss(
                 conf_model, fitresult, [x1 x2], _target; temp=temp, loss_matrix=loss_matrix
             )
         elseif plot_set_loss