diff --git a/dataset/__init__.py b/dataset/__init__.py
index 8ca08d80..e7b7b985 100644
--- a/dataset/__init__.py
+++ b/dataset/__init__.py
@@ -20,11 +20,15 @@
 from .wbc import WBC
 
 
+def mul255(x):
+    return x * 255
+
+
 def get_dataloader(args):
     transform_train = transforms.Compose([
         transforms.Resize((args.image_size,args.image_size)),
         transforms.ToTensor(),
-        transforms.Lambda(lambda x: x * 255)
+        transforms.Lambda(mul255)
     ])
 
     transform_train_seg = transforms.Compose([
@@ -35,7 +39,7 @@ def get_dataloader(args):
     transform_test = transforms.Compose([
         transforms.Resize((args.image_size, args.image_size)),
         transforms.ToTensor(),
-        transforms.Lambda(lambda x: x * 255)
+        transforms.Lambda(mul255)
     ])
 
     transform_test_seg = transforms.Compose([