diff --git a/web/pgadmin/setup/data_directory.py b/web/pgadmin/setup/data_directory.py
index eb710368b..6d39e276f 100644
--- a/web/pgadmin/setup/data_directory.py
+++ b/web/pgadmin/setup/data_directory.py
@@ -8,6 +8,7 @@
 ##########################################################################
 
 import os
+import sys
 import getpass
 from pgadmin.utils.constants import KERBEROS