diff --git a/scripts/restore b/scripts/restore
index 71ec97b1..154d47ca 100644
--- a/scripts/restore
+++ b/scripts/restore
@@ -47,7 +47,18 @@ ynh_restore_file --origin_path="/etc/nginx/conf.d/$domain.d/$app.conf"
 # RESTORE THE DATABASE
 #=================================================
 
-ynh_mysql_connect_as --user=$db_user --password=$db_pwd --database=$db_name < ./db.sql
+if [ $database = "mysql" ]; then
+	ynh_script_progression --message="Restoring  MySQL database..." --weight=1
+	db_pwd=$(ynh_app_setting_get --app=$app --key=mysqlpwd)
+	ynh_mysql_setup_db --db_user=$db_user --db_name=$db_name --db_pwd=$db_pwd
+	ynh_mysql_connect_as --user=$db_user --password=$db_pwd --database=$db_name < ./db.sql
+elif [ $database = "postgresql" ]; then
+	ynh_script_progression --message="Restoring the PostgreSQL database..." --weight=1
+	db_pwd=$(ynh_app_setting_get --app=$app --key=psqlpwd)
+	ynh_psql_test_if_first_run
+	ynh_psql_setup_db --db_user=$db_user --db_name=$db_name --db_pwd=$db_pwd
+	ynh_psql_execute_file_as_root --file="./db.sql" --database="$db_name"
+fi
 
 #=================================================
 # RESTORE VARIOUS FILES