diff --git a/src/model/edsr.py b/src/model/edsr.py
index 4ed986e2f310a965ce9519a1c7fafff5d81bf1ba..ef4ffb1e2f9c485140e92b3b06ddc93ab3877fba 100644
--- a/src/model/edsr.py
+++ b/src/model/edsr.py
@@ -23,7 +23,11 @@ class EDSR(nn.Module):
         kernel_size = 3 
         scale = args.scale[0]
         act = nn.ReLU(True)
-        self.url = url['r{}f{}x{}'.format(n_resblocks, n_feats, scale)]
+        url_name = 'r{}f{}x{}'.format(n_resblocks, n_feats, scale)
+        if url_name in url:
+            self.url = url[url_name]
+        else:
+            self.url = None
         self.sub_mean = common.MeanShift(args.rgb_range)
         self.add_mean = common.MeanShift(args.rgb_range, sign=1)