1 files changed, 4 insertions, 6 deletions
diff --git a/python/citco_submission.py b/python/citco_submission.py
index 2495c746..fcef1ac4 100644
--- a/python/citco_submission.py
+++ b/python/citco_submission.py
@@ -1,24 +1,22 @@
 from stat import S_ISREG
 import time
-import pickle
-from serenitas.utils.remote import SftpClient
 from citco_ops.utils import CitcoSubmission
 from paramiko.ssh_exception import SSHException
 
 
 def run():
-    sftp = CitcoSubmission.init_sftp()
+    CitcoSubmission.init_sftp()
     while True:
         try:
-            for f in sftp.client.listdir_iter():
+            for f in CitcoSubmission._sftp.client.listdir_iter():
                 if S_ISREG(f.st_mode):
                     CitcoSubmission.process(f.filename)
                     CitcoSubmission.check_cache()
 
                     CitcoSubmission.commit()
         except (SSHException, OSError):
-            sftp.client.close()
-            sftp = CitcoSubmission.init_sftp()
+            CitcoSubmission._sftp.client.close()
+            CitcoSubmission.init_sftp()
         time.sleep(60)