diff --git a/test/engines/erdos_reyni_gilbert_test.py b/test/engines/erdos_reyni_gilbert_test.py
index 139ae14b..b8270d2d 100644
--- a/test/engines/erdos_reyni_gilbert_test.py
+++ b/test/engines/erdos_reyni_gilbert_test.py
@@ -220,13 +220,13 @@
 >>> engine.gen_using(mtx_maker, 'testfile', {'vertices':10, 'probability':0.1, 'direct':0})
 5
 >>> g = mmread("testfile.mtx")
->>> print(list(g.data))
+>>> print(g.data.tolist())
 [1.0, 1.0, 1.0, 1.0, 1.0]
 >>> random.seed(4)
 >>> engine.gen_using(mtx_maker, 'testfile2', {'vertices':10, 'probability':0.1, 'direct':1})
 8
 >>> g = mmread("testfile2.mtx")
->>> print(list(g.data))
+>>> print(g.data.tolist())
 [1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0]
 >>> #################### tsv_maker ####################
 >>> random.seed(2)