diff --git a/modules/sd_hijack_directml.py b/modules/sd_hijack_directml.py
index b640fc848..c782921af 100644
--- a/modules/sd_hijack_directml.py
+++ b/modules/sd_hijack_directml.py
@@ -1,7 +1,7 @@
 import torch
 from tqdm.auto import tqdm
 
-from shared import device
+from modules.shared import device
 
 # k-diffusion
 from k_diffusion import sampling