diff --git a/src/data/dataset.py b/src/data/dataset.py
index 3874640..a9e77b7 100644
--- a/src/data/dataset.py
+++ b/src/data/dataset.py
@@ -30,10 +30,10 @@ class InpaintingData(Dataset):
             transforms.ColorJitter(0.05, 0.05, 0.05, 0.05),
             transforms.ToTensor()])
         self.mask_trans = transforms.Compose([
-            transforms.Resize(args.image_size, interpolation=transforms.InterpolationMode.NEAREST),
+            transforms.Resize(args.image_size),
             transforms.RandomHorizontalFlip(),
             transforms.RandomRotation(
-                (0, 45), interpolation=transforms.InterpolationMode.NEAREST),
+                (0, 45)),
         ])
 
         
@@ -77,4 +77,4 @@ if __name__ == '__main__':
     data = InpaintingData(args)
     print(len(data), len(data.mask_path))
     img, mask, filename = data[0]
-    print(img.size(), mask.size(), filename)
\ No newline at end of file
+    print(img.size(), mask.size(), filename)