diff --git a/quad/src/computation_graph/test/test_computation_graph.c b/quad/src/computation_graph/test/test_computation_graph.c
index e51880b4ac92d82d7d861e27124ff0a1f6c4ca59..30f977ed817dbea8d1125395988343857f3df25d 100644
--- a/quad/src/computation_graph/test/test_computation_graph.c
+++ b/quad/src/computation_graph/test/test_computation_graph.c
@@ -4,6 +4,8 @@
 #include "computation_graph.h"
 #include "graph_blocks.h"
 
+#define UNITY_INCLUDE_CONFIG_H
+
 #define GRAPH_TEST_EPS 0.00001
 
 
@@ -224,9 +226,7 @@ void test_that_new_nodes_can_be_created_by_ID() {
     struct computation_graph *graph = create_graph();
     int desired_id = 87;
     int add_block = graph_add_node_id(graph, desired_id, "Add", &node_add_type);
-    if (add_block != desired_id) {
-        return -1;
-    }
+    TEST_ASSERT_EQUAL_INT(desired_id, add_block);
     int const1 = graph_add_node_id(graph, 12, "const1", &node_const_type);
     graph_set_param_val(graph, const1, CONST_SET, 3.5);
     int const2 = graph_add_node_id(graph, 123, "const2", &node_const_type);