diff --git a/test/unit_tests_qedmodel.jl b/test/unit_tests_qedmodel.jl
index 356e6af..af63fdf 100644
--- a/test/unit_tests_qedmodel.jl
+++ b/test/unit_tests_qedmodel.jl
@@ -279,9 +279,13 @@ end
 
     insert_edge!(graph, sum_node, d_exit, track = false)
 
-    compton_function = get_compute_function(graph, process, machine)
-
     input = [gen_process_input(process) for _ in 1:1000]
 
+    compton_function = get_compute_function(graph, process, machine)
+    @test isapprox(compton_function.(input), compton_groundtruth.(input))
+
+    graph_generated = gen_graph(process)
+
+    compton_function = get_compute_function(graph_generated, process, machine)
     @test isapprox(compton_function.(input), compton_groundtruth.(input))
 end