diff --git a/celeba_make_dataset.ipynb b/celeba_make_dataset.ipynb
index ff312f5..22d8c6c 100644
--- a/celeba_make_dataset.ipynb
+++ b/celeba_make_dataset.ipynb
@@ -414,7 +414,7 @@
    "source": [
     "with h5py.File(''.join(['datasets/faces_dataset_new.h5']), 'w') as f:\n",
     "    dset_face = f.create_dataset(\"images\", data = images)\n",
-    "    dset_headers = f.create_dataset('headers', data = headers)\n",
+    "    dset_headers = f.create_dataset('headers', data = np.string_(headers))\n",
     "    dset_label_input = f.create_dataset('label_input', data = label_input)"
    ]
   },