diff --git a/test/fitting/test-fit-multi.jl b/test/fitting/test-fit-multi.jl
index 4ac1a0bf..870bccd4 100644
--- a/test/fitting/test-fit-multi.jl
+++ b/test/fitting/test-fit-multi.jl
@@ -5,7 +5,7 @@ include("../dummies.jl")
 
 # generate some fake powerlaw data with three components
 dummy_data =
-    make_dummy_dataset((E) -> (E^(-0.1) + E^(-3.0) + E^(-1.0)); units = "count / s")
+    make_dummy_dataset((E) -> (E^(-0.1) + E^(-3.0) + E^(-1.0)); units = "count / s keV")
 
 # model with two components
 model = PowerLaw() + PowerLaw()
diff --git a/test/fitting/test-fit-optim.jl b/test/fitting/test-fit-optim.jl
index b459e702..6e207c89 100644
--- a/test/fitting/test-fit-optim.jl
+++ b/test/fitting/test-fit-optim.jl
@@ -6,7 +6,7 @@ include("../dummies.jl")
 
 # generate some fake powerlaw data with three components
 dummy_data =
-    make_dummy_dataset((E) -> (E^(-0.1) + E^(-3.0) + E^(-1.0)); units = "count / s")
+    make_dummy_dataset((E) -> (E^(-0.1) + E^(-3.0) + E^(-1.0)); units = "count / s keV")
 
 # model with two components
 model = PowerLaw(K = FitParam(10.0)) + PowerLaw(K = FitParam(10.0))
diff --git a/test/fitting/test-fit-powerlaw.jl b/test/fitting/test-fit-powerlaw.jl
index b4fd7ef1..7644c95a 100644
--- a/test/fitting/test-fit-powerlaw.jl
+++ b/test/fitting/test-fit-powerlaw.jl
@@ -4,7 +4,7 @@ using SpectralFitting
 include("../dummies.jl")
 
 # generate some fake powerlaw data
-dummy_data = make_dummy_dataset((E) -> (E^(-3.0)); units = "count / s")
+dummy_data = make_dummy_dataset((E) -> (E^(-3.0)); units = "count / s keV")
 
 # test that both julia and xspec implementations can fit simple
 model = PowerLaw()
@@ -22,7 +22,7 @@ result = fit(prob, LevenbergMarquadt())
 @test result.u ≈ [12.06629478087094, 3.080992500319396] atol = 1e-4
 
 # composite power law
-dummy_data = make_dummy_dataset((E) -> (E^(-3.0) + E^(-1)); units = "count / s")
+dummy_data = make_dummy_dataset((E) -> (E^(-3.0) + E^(-1)); units = "count / s keV")
 
 # julia implementation
 model = PowerLaw() + PowerLaw(a = FitParam(1.0))