diff --git a/dan/predict/prediction.py b/dan/predict/prediction.py
index 09e7a7ba9f34021a64cce36577bf30c8cfaac425..30c9ff5f8382759b66567ee3b88fb2bfb554a4ab 100644
--- a/dan/predict/prediction.py
+++ b/dan/predict/prediction.py
@@ -2,6 +2,7 @@
 
 import os
 import pickle
+from pathlib import Path
 
 import cv2
 import numpy as np
@@ -100,6 +101,7 @@ class DAN:
         extract_objects=False,
         word_separators=["\n", " "],
         line_separators=["\n"],
+        start_token=None,
     ):
         """
         Run prediction on an input image.
@@ -352,4 +354,4 @@ def run(
 
     json_filename = f"{output}/{image.stem}.json"
     logger.info(f"Saving JSON prediction in {json_filename}")
-    save_json(json_filename, result)
+    save_json(Path(json_filename), result)