diff --git a/src/components/instant_gradients.jl b/src/components/instant_gradients.jl
index cc2801c6e5f77b4860bb9e06f1c5ca84f879111a..7c71246607432f645420f9c6e2080cdee29d2700 100644
--- a/src/components/instant_gradients.jl
+++ b/src/components/instant_gradients.jl
@@ -26,7 +26,7 @@ To have a variable `orientation`, see [`InstantGradient3D`](@ref).
 - [`qval`](@ref): Spatial frequency on which spins will be dephased due to this pulsed gradient in rad/um.
 - [`spoiler_scale`](@ref): Length-scale on which spins will be dephased by exactly 2Ï€ in mm.
 """
-struct InstantGradient1D <: GradientBlock
+struct InstantGradient1D <: InstantGradient
     qval :: VariableType
     orientation :: SVector{3, Number}
     group :: Union{Nothing, Symbol}
@@ -57,7 +57,7 @@ To have a fixed `orientation`, see [`InstantGradient1D`](@ref).
 - [`qval`](@ref): Norm of spatial frequency on which spins will be dephased due to this pulsed gradient in rad/um.
 - [`spoiler_scale`](@ref): Vector with length-scale on which spins will be dephased by exactly 2Ï€ in mm.
 """
-struct InstantGradient3D <: GradientBlock
+struct InstantGradient3D <: InstantGradient
     qvec :: SVector{3, VariableType}
     group :: Union{Nothing, Symbol}
 end