diff --git a/clearml_session/interactive_session_task.py b/clearml_session/interactive_session_task.py
index 95d0017..416c9bd 100644
--- a/clearml_session/interactive_session_task.py
+++ b/clearml_session/interactive_session_task.py
@@ -14,6 +14,7 @@ import requests
 from clearml import Task, StorageManager
 from clearml.backend_api import Session
 from pathlib2 import Path
+from tcp_proxy import TcpProxy
 
 # noinspection SpellCheckingInspection
 default_ssh_fingerprint = {