diff --git a/src/aiunn/finetune.py b/src/aiunn/finetune.py
index 3f7b941..d785030 100644
--- a/src/aiunn/finetune.py
+++ b/src/aiunn/finetune.py
@@ -12,7 +12,7 @@ from torchvision import transforms
 from tqdm import tqdm
 
 from aiia import AIIABase
-from aiunn.upsampler import Upsampler
+from upsampler import Upsampler
 
 # Define a simple EarlyStopping class to monitor the epoch loss.
 class EarlyStopping: