diff --git a/test/OmicsIntegrator2/input/oi2-edges.txt b/test/OmicsIntegrator2/input/oi2-edges.txt
index f56173e1..263fbd4a 100644
--- a/test/OmicsIntegrator2/input/oi2-edges.txt
+++ b/test/OmicsIntegrator2/input/oi2-edges.txt
@@ -1,6 +1,10 @@
 protein1	protein2	cost
-A	B	0.9
-A	C	0.1
-B	D	0.9
-C	D	0.1
-A	D	10
+A	B	0.52
+B	C	0.73
+A	D	1.38
+C	D	0.61
+C	E	0.91
+C	F	1.0
+F	G	0.74
+G	H	0.58
+G	I	0.84
diff --git a/test/OmicsIntegrator2/input/oi2-prizes.txt b/test/OmicsIntegrator2/input/oi2-prizes.txt
index fd4250fb..90d09400 100644
--- a/test/OmicsIntegrator2/input/oi2-prizes.txt
+++ b/test/OmicsIntegrator2/input/oi2-prizes.txt
@@ -1,3 +1,3 @@
-name	 prize
-A	10.0
-D	10.0
+name	prize
+C	5.7
+A	2.0
diff --git a/test/OmicsIntegrator2/test_oi2.py b/test/OmicsIntegrator2/test_oi2.py
index b0ba3ce3..61febe6a 100644
--- a/test/OmicsIntegrator2/test_oi2.py
+++ b/test/OmicsIntegrator2/test_oi2.py
@@ -20,7 +20,9 @@ def test_oi2_required(self):
         OUT_FILE.unlink(missing_ok=True)
         OmicsIntegrator2.run(edges=EDGE_FILE,
                              prizes=PRIZE_FILE,
-                             output_file=OUT_FILE)
+                             output_file=OUT_FILE,
+                             b=2,
+                             g=0)
         assert OUT_FILE.exists()
 
     def test_oi2_some_optional(self):