diff --git a/src/jop_prop2DAcoIsoDenQ_DEO2_FDTD.jl b/src/jop_prop2DAcoIsoDenQ_DEO2_FDTD.jl
index 5fba07d..88a141f 100644
--- a/src/jop_prop2DAcoIsoDenQ_DEO2_FDTD.jl
+++ b/src/jop_prop2DAcoIsoDenQ_DEO2_FDTD.jl
@@ -495,7 +495,7 @@ function JopProp2DAcoIsoDenQ_DEO2_FDTD_nonlinearforward!(d::AbstractArray, m::Ab
             end
 
             # scale spatial derivatives by v^2/b to make them temporal derivatives
-            cumtime_io += @elapsed WaveFD.scale_spatial_derivatives!(p)
+            cumtime_pr += @elapsed WaveFD.scale_spatial_derivatives!(p)
 
             if kwargs[:srcfieldfile] != ""
                 cumtime_io += @elapsed for active_wavefield in active_wavefields
diff --git a/src/jop_prop2DAcoTTIDenQ_DEO2_FDTD.jl b/src/jop_prop2DAcoTTIDenQ_DEO2_FDTD.jl
index 37b0e78..5820b40 100644
--- a/src/jop_prop2DAcoTTIDenQ_DEO2_FDTD.jl
+++ b/src/jop_prop2DAcoTTIDenQ_DEO2_FDTD.jl
@@ -537,7 +537,7 @@ function JopProp2DAcoTTIDenQ_DEO2_FDTD_nonlinearforward!(d::AbstractArray, m::Ab
             end
 
             # scale spatial derivatives by v^2/b to make them temporal derivatives
-            cumtime_io += @elapsed WaveFD.scale_spatial_derivatives!(p)
+            cumtime_pr += @elapsed WaveFD.scale_spatial_derivatives!(p)
 
             if kwargs[:srcfieldfile] != ""
                 cumtime_io += @elapsed for active_wavefield in active_wavefields
diff --git a/src/jop_prop2DAcoVTIDenQ_DEO2_FDTD.jl b/src/jop_prop2DAcoVTIDenQ_DEO2_FDTD.jl
index 294b6cf..2f1ba51 100644
--- a/src/jop_prop2DAcoVTIDenQ_DEO2_FDTD.jl
+++ b/src/jop_prop2DAcoVTIDenQ_DEO2_FDTD.jl
@@ -534,7 +534,7 @@ function JopProp2DAcoVTIDenQ_DEO2_FDTD_nonlinearforward!(d::AbstractArray, m::Ab
             end
 
             # scale spatial derivatives by v^2/b to make them temporal derivatives
-            cumtime_io += @elapsed WaveFD.scale_spatial_derivatives!(p)
+            cumtime_pr += @elapsed WaveFD.scale_spatial_derivatives!(p)
 
             if kwargs[:srcfieldfile] != ""
                 cumtime_io += @elapsed for active_wavefield in active_wavefields
diff --git a/src/jop_prop3DAcoIsoDenQ_DEO2_FDTD.jl b/src/jop_prop3DAcoIsoDenQ_DEO2_FDTD.jl
index ef00733..e959a53 100644
--- a/src/jop_prop3DAcoIsoDenQ_DEO2_FDTD.jl
+++ b/src/jop_prop3DAcoIsoDenQ_DEO2_FDTD.jl
@@ -523,7 +523,7 @@ function JopProp3DAcoIsoDenQ_DEO2_FDTD_nonlinearforward!(d::AbstractArray, m::Ab
             end
 
             # scale spatial derivatives by v^2/b to make them temporal derivatives
-            cumtime_io += @elapsed WaveFD.scale_spatial_derivatives!(p)
+            cumtime_pr += @elapsed WaveFD.scale_spatial_derivatives!(p)
 
             if kwargs[:srcfieldfile] != ""
                 cumtime_io += @elapsed for active_wavefield in active_wavefields
diff --git a/src/jop_prop3DAcoTTIDenQ_DEO2_FDTD.jl b/src/jop_prop3DAcoTTIDenQ_DEO2_FDTD.jl
index b2f5e65..12ba495 100644
--- a/src/jop_prop3DAcoTTIDenQ_DEO2_FDTD.jl
+++ b/src/jop_prop3DAcoTTIDenQ_DEO2_FDTD.jl
@@ -575,7 +575,7 @@ function JopProp3DAcoTTIDenQ_DEO2_FDTD_nonlinearforward!(d::AbstractArray, m::Ab
             end
 
             # scale spatial derivatives by v^2/b to make them temporal derivatives
-            cumtime_io += @elapsed WaveFD.scale_spatial_derivatives!(p)
+            cumtime_ex += @elapsed WaveFD.scale_spatial_derivatives!(p)
 
             if kwargs[:srcfieldfile] != ""
                 cumtime_io += @elapsed for active_wavefield in active_wavefields
diff --git a/src/jop_prop3DAcoVTIDenQ_DEO2_FDTD.jl b/src/jop_prop3DAcoVTIDenQ_DEO2_FDTD.jl
index 754852a..98668b1 100644
--- a/src/jop_prop3DAcoVTIDenQ_DEO2_FDTD.jl
+++ b/src/jop_prop3DAcoVTIDenQ_DEO2_FDTD.jl
@@ -560,7 +560,7 @@ function JopProp3DAcoVTIDenQ_DEO2_FDTD_nonlinearforward!(d::AbstractArray, m::Ab
             end
 
             # scale spatial derivatives by v^2/b to make them temporal derivatives
-            cumtime_io += @elapsed WaveFD.scale_spatial_derivatives!(p)
+            cumtime_pr += @elapsed WaveFD.scale_spatial_derivatives!(p)
 
             if kwargs[:srcfieldfile] != ""
                 cumtime_io += @elapsed for active_wavefield in active_wavefields