diff --git a/arelight/run/infer.py b/arelight/run/infer.py
index a223040..e4a459b 100644
--- a/arelight/run/infer.py
+++ b/arelight/run/infer.py
@@ -319,7 +319,7 @@ def __entity_display_value(entity_value):
     settings.append({
         "labels_scaler": labels_scaler,
         # We provide this settings for inference.
-        "predict_filepath": collection_target_func(DataType.Test),
+        "predict_filepath": collection_target_func(DataType.Test) + predict_extension[args.inference_writer],
         "samples_io": sampling_engines_setup["arekit"]["samples_io"],
         "predict_reader": predict_readers[args.inference_writer]
     })