diff --git a/core/src/validation.jl b/core/src/validation.jl
index cafabd4db..4f422a5d6 100644
--- a/core/src/validation.jl
+++ b/core/src/validation.jl
@@ -321,7 +321,13 @@ sort_by_function(table::StructVector{<:Legolas.AbstractRecord}) = sort_by_id
 sort_by_function(table::StructVector{TabulatedRatingCurveStaticV1}) = sort_by_id_state_level
 sort_by_function(table::StructVector{BasinProfileV1}) = sort_by_id_level
 
-const TimeSchemas = Union{TabulatedRatingCurveTimeV1, FlowBoundaryTimeV1, BasinForcingV1}
+const TimeSchemas = Union{
+    BasinForcingV1,
+    FlowBoundaryTimeV1,
+    LevelBoundaryTimeV1,
+    PidControlTimeV1,
+    TabulatedRatingCurveTimeV1,
+}
 
 function sort_by_function(table::StructVector{<:TimeSchemas})
     return sort_by_time_id