diff --git a/EpiAware/test/test_models.jl b/EpiAware/test/test_models.jl
index 9e732c287..4cc32b5f5 100644
--- a/EpiAware/test/test_models.jl
+++ b/EpiAware/test/test_models.jl
@@ -32,7 +32,7 @@
     gen = generated_quantities(test_mdl, rand(test_mdl))
 
     #Check model sampled
-    @test eltype(gen.generated_y_t) <: Integer
+    @test eltype(gen.generated_y_t) <: AbstractFloat
     @test eltype(gen.I_t) <: AbstractFloat
     @test length(gen.I_t) == time_horizon
 end
@@ -71,7 +71,7 @@ end
     gens = generated_quantities(test_mdl, chn)
 
     #Check model sampled
-    @test eltype(gens[1].generated_y_t) <: Integer
+    @test eltype(gens[1].generated_y_t) <: AbstractFloat
     @test eltype(gens[1].I_t) <: AbstractFloat
     @test length(gens[1].I_t) == time_horizon
 end
@@ -110,7 +110,7 @@ end
     gens = generated_quantities(test_mdl, chn)
 
     #Check model sampled
-    @test eltype(gens[1].generated_y_t) <: Integer
+    @test eltype(gens[1].generated_y_t) <: AbstractFloat
     @test eltype(gens[1].I_t) <: AbstractFloat
     @test length(gens[1].I_t) == time_horizon
 end