diff --git a/scripts/restore b/scripts/restore
index c39c75f..181c9f1 100644
--- a/scripts/restore
+++ b/scripts/restore
@@ -27,7 +27,7 @@ ynh_script_progression --message="Restoring the app main directory..." --weight=
 ynh_restore_file --origin_path="$install_dir"
 
 chown -R root: $install_dir/
-chown -R $admin: $install_dir/.venv/
+chown -R $admin: $install_dir/venv/
 
 #=================================================
 # SPECIFIC RESTORATION
diff --git a/scripts/upgrade b/scripts/upgrade
index 8c10b2e..483627f 100644
--- a/scripts/upgrade
+++ b/scripts/upgrade
@@ -56,7 +56,7 @@ then
 	pushd $install_dir
 		python3 -m venv $install_dir/venv
 		source $install_dir/venv/bin/activate
-		ynh_exec_warn_less pip install npm@latest jupyterlab==$jupyterlab_version jupyterhub notebook jupyter_collaboration jupyterhub-ldapauthenticator pyzmq
+		ynh_exec_warn_less pip install npm@latest jupyterlab==$jupyterlab_version jupyterhub notebook jupyter_collaboration jupyterhub-ldapauthenticator pyzmq toml
 	popd
 fi