diff --git a/desktop_env/controllers/setup.py b/desktop_env/controllers/setup.py
index 7604d11..b4ae4b2 100644
--- a/desktop_env/controllers/setup.py
+++ b/desktop_env/controllers/setup.py
@@ -2,6 +2,7 @@ import json
 import logging
 import os
 import os.path
+import platform
 import shutil
 import sqlite3
 import tempfile