diff --git a/lesson6-keyboard-commands/TelloKeyboardCommands.ipynb b/lesson6-keyboard-commands/TelloKeyboardCommands.ipynb
index 74d8de0..05c2c3f 100644
--- a/lesson6-keyboard-commands/TelloKeyboardCommands.ipynb
+++ b/lesson6-keyboard-commands/TelloKeyboardCommands.ipynb
@@ -10,6 +10,7 @@
     "import socket\n",
     "import threading\n",
     "import time\n",
+    "import sys\n",
     "\n",
     "# IP and port of Tello\n",
     "tello_address = ('192.168.10.1', 8889)\n",
diff --git a/lesson6-keyboard-commands/TelloKeyboardCommands.py b/lesson6-keyboard-commands/TelloKeyboardCommands.py
index f5be313..ab3a9e8 100644
--- a/lesson6-keyboard-commands/TelloKeyboardCommands.py
+++ b/lesson6-keyboard-commands/TelloKeyboardCommands.py
@@ -6,6 +6,7 @@
 import socket
 import threading
 import time
+import sys
 
 # IP and port of Tello
 tello_address = ('192.168.10.1', 8889)