Fix when using key_dir config