diff --git a/pc-client/main.py b/pc-client/main.py
index d895030..58b07bb 100644
--- a/pc-client/main.py
+++ b/pc-client/main.py
@@ -5,6 +5,7 @@ import threading
 import json
 try:
     import pystray
+    from pystray import Icon as icon, Menu as menu, MenuItem as item
     from PIL import Image, ImageDraw
     import serial
     import win32api