diff --git a/tests/source_SCC_test.py b/tests/source_SCC_test.py
index e0a0c0f..f308e02 100644
--- a/tests/source_SCC_test.py
+++ b/tests/source_SCC_test.py
@@ -29,7 +29,7 @@ def test_find_source_nodes():
     assert source_nodes == ["source"]
 
     perc_space = percolate_space(graph, {})
-    perc_bn = percolate_network(bn, perc_space, ctx=graph)
+    perc_bn = percolate_network(bn, perc_space, symbolic_network=graph)
 
     source_nodes = find_source_nodes(perc_bn)
 
diff --git a/tests/space_utils_test.py b/tests/space_utils_test.py
index 643d11b..7c37851 100644
--- a/tests/space_utils_test.py
+++ b/tests/space_utils_test.py
@@ -98,13 +98,21 @@ def test_network_percolation():
     )
     graph = AsynchronousGraph(bn)
 
-    percolated_bn = percolate_network(bn, percolate_space(graph, {"c": 0}), ctx=graph)
-    percolated_bn = percolate_network(bn, percolate_space(graph, {"c": 0}), ctx=graph)
+    percolated_bn = percolate_network(
+        bn, percolate_space(graph, {"c": 0}), symbolic_network=graph
+    )
+    percolated_bn = percolate_network(
+        bn, percolate_space(graph, {"c": 0}), symbolic_network=graph
+    )
     assert "false" == str(percolated_bn.get_update_function("c"))
     assert "false" == str(percolated_bn.get_update_function("a"))
     assert "true" == str(percolated_bn.get_update_function("b"))
-    percolated_bn = percolate_network(bn, percolate_space(graph, {"c": 1}), ctx=graph)
-    percolated_bn = percolate_network(bn, percolate_space(graph, {"c": 1}), ctx=graph)
+    percolated_bn = percolate_network(
+        bn, percolate_space(graph, {"c": 1}), symbolic_network=graph
+    )
+    percolated_bn = percolate_network(
+        bn, percolate_space(graph, {"c": 1}), symbolic_network=graph
+    )
     assert "true" == str(percolated_bn.get_update_function("c"))
     assert "b" == str(percolated_bn.get_update_function("a"))
     assert "!a" == str(percolated_bn.get_update_function("b"))