diff --git a/src/ai/backend/common/types.py b/src/ai/backend/common/types.py
index <HASH>..<HASH> 100644
--- a/src/ai/backend/common/types.py
+++ b/src/ai/backend/common/types.py
@@ -18,6 +18,8 @@ import uuid
 
 import attr
 
+from .docker import ImageRef
+
 __all__ = (
     'aobject',
     'DeviceId',
@@ -580,6 +582,7 @@ class KernelCreationConfig(TypedDict):
     environ: Mapping[str, str]
     mounts: Sequence[str]              # list of mount expressions
     mount_map: Mapping[str, str]       # Mapping of vfolder custom mount path
+    package_directory: Sequence[str]
     idle_timeout: int
     bootstrap_script: Optional[str]
     startup_command: Optional[str]
Add package_directory to KernelCreationConfig