diff --git a/libavformat/hlsenc.c b/libavformat/hlsenc.c
index 0ea93480a5..6d573db094 100644
--- a/libavformat/hlsenc.c
+++ b/libavformat/hlsenc.c
@@ -1443,6 +1443,7 @@ static int hls_write_header(AVFormatContext *s)
         }
     }
 
+    if (hls->segment_type == SEGMENT_TYPE_FMP4) {
     if (av_strcasecmp(hls->fmp4_init_filename, "init.mp4")) {
         hls->base_output_dirname = av_malloc(fmp4_init_filename_len);
         if (!hls->base_output_dirname) {
@@ -1472,6 +1473,7 @@ static int hls_write_header(AVFormatContext *s)
             av_strlcpy(hls->base_output_dirname, hls->fmp4_init_filename, fmp4_init_filename_len);
         }
     }
+    }
 
     if (!hls->use_localtime) {
         ret = sls_flag_check_duration_size_index(hls);