diff --git a/studio/backend/main.py b/studio/backend/main.py
index b29fdb3908..2d00f985e5 100644
--- a/studio/backend/main.py
+++ b/studio/backend/main.py
@@ -3,6 +3,7 @@ Main FastAPI application for Unsloth UI Backend
 """
 import os
 import secrets
+import shutil
 from contextlib import asynccontextmanager
 
 from fastapi import FastAPI