diff --git a/web/pgadmin/tools/backup/__init__.py b/web/pgadmin/tools/backup/__init__.py
index 9d4fa4c31..fbec8396a 100644
--- a/web/pgadmin/tools/backup/__init__.py
+++ b/web/pgadmin/tools/backup/__init__.py
@@ -25,6 +25,7 @@ from config import PG_DEFAULT_DRIVER
 # This unused import is required as API test cases will fail if we remove it,
 # Have to identify the cause and then remove it.
 from pgadmin.model import Server, SharedServer
+from flask_security import current_user
 from pgadmin.misc.bgprocess import escape_dquotes_process_arg
 from pgadmin.utils.constants import MIMETYPE_APP_JS, SERVER_NOT_FOUND
 from pgadmin.tools.grant_wizard import get_data
diff --git a/web/pgadmin/tools/restore/__init__.py b/web/pgadmin/tools/restore/__init__.py
index 45eb2c85b..cb5683648 100644
--- a/web/pgadmin/tools/restore/__init__.py
+++ b/web/pgadmin/tools/restore/__init__.py
@@ -13,6 +13,9 @@ import json
 
 from flask import render_template, request, current_app, Response
 from flask_babel import gettext as _
+# This unused import is required as API test cases will fail if we remove it,
+# Have to identify the cause and then remove it.
+from flask_security import current_user
 from pgadmin.user_login_check import pga_login_required
 from pgadmin.misc.bgprocess.processes import BatchProcess, IProcessDesc
 from pgadmin.utils import PgAdminModule, fs_short_path, does_utility_exist, \