diff --git a/figures/SchueckerSchmidt2017/utils.py b/figures/SchueckerSchmidt2017/utils.py
index 506fc4f06208ef6de9a3ec1e120c6eca4576e5b3..556bd576a20b18cccfc7b186be5815a31e7be6c6 100644
--- a/figures/SchueckerSchmidt2017/utils.py
+++ b/figures/SchueckerSchmidt2017/utils.py
@@ -1,5 +1,4 @@
 import copy
-import data_wrapper.data_wrapper as dw
 import multiprocessing as mp
 import numpy as np
 import os
@@ -90,7 +89,8 @@ def save_iteration(step, data):
         os.mkdir(data_dir)
     except FileExistsError:
         pass
-    dw.save('iteration_{}'.format(step), data)
+    for key in ['parameters', 'K_prime', 'results']:
+        np.save('iteration_{}/{}.npy'.format(step, key), data[key])
 
 
 def load_iteration(step):