diff --git a/tests/run_unit.jl b/tests/run_unit.jl
index e67bfff5bc391c1ef70d380a5dc7f51832b693ab..46dc9d94fdcc86477a987a055631c8d550090075 100644
--- a/tests/run_unit.jl
+++ b/tests/run_unit.jl
@@ -11,8 +11,10 @@ using Test
     @test include("unit/create_automata.jl")
     @test include("unit/create_models.jl")
     
+    @test include("unit/density_pm.jl")
     @test include("unit/dist_lp.jl")
     @test include("unit/dist_lp_var.jl")
+    @test include("unit/draw_pm.jl")
     
     @test include("unit/getindex_access_trajectory.jl")
     @test include("unit/is_always_bounded_sir.jl")