diff --git a/src/all_building_blocks/wait_blocks.jl b/src/all_building_blocks/wait_blocks.jl
index 3558cf818b12aa6e03677a1f5146fa2fb8c26f43..ce8615f2517cba468f8f971dd93e2464955ddd87 100644
--- a/src/all_building_blocks/wait_blocks.jl
+++ b/src/all_building_blocks/wait_blocks.jl
@@ -5,7 +5,7 @@ import ...Variables: get_free_variable, VariableType, duration
 import ...Components: NoGradient
 import ..BaseBuildingBlocks: BaseBuildingBlock
 
-struct Wait <: BuildingBlock
+struct Wait <: BaseBuildingBlock
     duration :: VariableType
     function Wait(var)
         res = new(get_free_variable(var))