diff --git a/misc/re-optimize-corpora.py b/misc/re-optimize-corpora.py
index 3f247c58b0..f481821d9a 100755
--- a/misc/re-optimize-corpora.py
+++ b/misc/re-optimize-corpora.py
@@ -18,4 +18,7 @@
     corpora = f.read().splitlines()
 
 for c in corpora:
-    subprocess.run([args.annis, args.data_dir, "-c", "corpus " + c ,  "-c", "re-optimize"])
\ No newline at end of file
+    if args.disk_based:
+        subprocess.run([args.annis, args.data_dir, "-c", "set-disk-based true", "-c", "corpus " + c ,  "-c", "re-optimize"])
+    else:
+        subprocess.run([args.annis, args.data_dir, "-c", "corpus " + c ,  "-c", "re-optimize"])