diff --git a/src/aiunn/finetune.py b/src/aiunn/finetune.py
index 9729208..3f7b941 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.upsample import Upsampler
+from aiunn.upsampler import Upsampler
 
 # Define a simple EarlyStopping class to monitor the epoch loss.
 class EarlyStopping:
diff --git a/src/aiunn/upsample.py b/src/aiunn/upsampler.py
similarity index 100%
rename from src/aiunn/upsample.py
rename to src/aiunn/upsampler.py