diff --git a/scripts/restore b/scripts/restore
index 57c95b0..012a214 100644
--- a/scripts/restore
+++ b/scripts/restore
@@ -91,7 +91,8 @@ ynh_install_extra_app_dependencies --repo="https://repo.jellyfin.org/debian $( l
 #=================================================
 ynh_script_progression --message="Restoring the systemd configuration..." --time --weight=1
 
-ynh_restore_file --origin_path="/etc/systemd/system/$app.service"
+ynh_restore_file --origin_path="/lib/systemd/system/jellyfin.service"
+ynh_restore_file --origin_path="/etc/systemd/system/jellyfin.service.d"
 systemctl enable $app.service
 
 #=================================================
diff --git a/scripts/upgrade b/scripts/upgrade
index 744dc15..ee3523c 100644
--- a/scripts/upgrade
+++ b/scripts/upgrade
@@ -18,6 +18,7 @@ app=$YNH_APP_INSTANCE_NAME
 domain=$(ynh_app_setting_get --app=$app --key=domain)
 path_url=$(ynh_app_setting_get --app=$app --key=path)
 final_path=$(ynh_app_setting_get --app=$app --key=final_path)
+port=$(ynh_app_setting_get --app=$app --key=port)
 
 #=================================================
 # CHECK VERSION