diff --git a/ohmpi/ohmpi.py b/ohmpi/ohmpi.py
index 20e7f6ab7d593850dc477228e081774c53f2b6a9..b21db3795f53b1098b4a8a2777062ddab891191a 100644
--- a/ohmpi/ohmpi.py
+++ b/ohmpi/ohmpi.py
@@ -723,7 +723,7 @@ class OhmPi(object):
                 for i, row in enumerate(x[1:]):
                     xs = row.split(',')
                     f.write(','.join(xs[:icol]))
-                    fw = np.array([icol:])
+                    fw = np.array(xs[icol:])
                     fw_pad = fw.reshape((5, -1))
                     fw_padded = np.zeros((max_length, 5))
                     fw_padded[:fw_pad.shape[0], :] = fw_pad