diff --git a/astrbot/core/computer/booters/shipyard.py b/astrbot/core/computer/booters/shipyard.py
index 77f31bc40..806bf611b 100644
--- a/astrbot/core/computer/booters/shipyard.py
+++ b/astrbot/core/computer/booters/shipyard.py
@@ -1,7 +1,8 @@
 from __future__ import annotations
 
+import functools
 import shlex
-from typing import Any
+from typing import TYPE_CHECKING, Any
 
 from shipyard import ShipyardClient, Spec