diff --git a/tools/lycoris_locon_extract.py b/tools/lycoris_locon_extract.py
index b842faf..cd58761 100644
--- a/tools/lycoris_locon_extract.py
+++ b/tools/lycoris_locon_extract.py
@@ -124,8 +124,8 @@ ARGS = get_args()
 
 
 from lycoris.utils import extract_diff
-from lycoris.kohya.model_utils import load_models_from_stable_diffusion_checkpoint
-from lycoris.kohya.sdxl_model_util import load_models_from_sdxl_checkpoint
+from library.model_util import load_models_from_stable_diffusion_checkpoint
+from library.sdxl_model_util import load_models_from_sdxl_checkpoint
 
 import torch
 from safetensors.torch import save_file
@@ -187,4 +187,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
\ No newline at end of file
+    main()