diff --git a/aimodel/src/subcommands/pretrain_predict.py b/aimodel/src/subcommands/pretrain_predict.py
index 716ebe3..9577aa3 100644
--- a/aimodel/src/subcommands/pretrain_predict.py
+++ b/aimodel/src/subcommands/pretrain_predict.py
@@ -8,7 +8,7 @@ import re
 from loguru import logger
 import tensorflow as tf
 import numpy as np
-from aimodel.src.lib.io.writefile import writefile
+from lib.io.writefile import writefile
 
 from lib.io.handle_open import handle_open
 from lib.ai.RainfallWaterContraster import RainfallWaterContraster