diff --git a/config/setting.py b/config/setting.py
index 6db5d4f..e905304 100644
--- a/config/setting.py
+++ b/config/setting.py
@@ -24,7 +24,7 @@
 
 # model.networks.sr
 SR_IMAGE_TYPE = "RGB"
-SR_IMAGE_CHANNEL = image_type_choices[IMAGE_TYPE]
+SR_IMAGE_CHANNEL = image_type_choices[SR_IMAGE_TYPE]
 
 # Data processing
 # ****** torchvision.transforms.Compose ******