From 09f23e4eff131fd64016b50d064e267b9dfb0d79 Mon Sep 17 00:00:00 2001
From: Jose Daniel Lara <jdlara@berkeley.edu>
Date: Mon, 6 May 2024 21:14:36 -0600
Subject: [PATCH] more fixes for tests

---
 src/core/optimization_container.jl                       | 2 +-
 src/devices_models/devices/common/objective_functions.jl | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/core/optimization_container.jl b/src/core/optimization_container.jl
index 75136c85b8..5cd63b65b3 100644
--- a/src/core/optimization_container.jl
+++ b/src/core/optimization_container.jl
@@ -376,6 +376,7 @@ function check_optimization_container(container::OptimizationContainer)
             error("The model container has invalid values in $(encode_key_as_string(k))")
         end
     end
+    container.settings_copy = copy_for_serialization(container.settings)
     return
 end
 
@@ -620,7 +621,6 @@ function build_impl!(
         LOG_GROUP_OPTIMIZATION_CONTAINER
 
     check_optimization_container(container)
-
     return
 end
 
diff --git a/src/devices_models/devices/common/objective_functions.jl b/src/devices_models/devices/common/objective_functions.jl
index baa63a2298..f43a2e394f 100644
--- a/src/devices_models/devices/common/objective_functions.jl
+++ b/src/devices_models/devices/common/objective_functions.jl
@@ -625,7 +625,7 @@ end
 function _add_pwl_term!(
     container::OptimizationContainer,
     component::T,
-    cost_data::AbstractVector{PSY.PiecewiseLinearData},
+    cost_data::AbstractVector{PSY.PiecewiseStepData},
     ::U,
     ::V,
 ) where {T <: PSY.Component, U <: VariableType, V <: AbstractServiceFormulation}