diff --git a/INetwork.py b/INetwork.py
index b51c432..833b36a 100644
--- a/INetwork.py
+++ b/INetwork.py
@@ -189,7 +189,7 @@ def preprocess_image(image_path, load_dims=False, read_mode="color"):
         aspect_ratio = float(img_HEIGHT) / img_WIDTH
 
         img_width = args.img_size
-        if args.maintain_aspect_ratio:
+        if maintain_aspect_ratio:
             img_height = int(img_width * aspect_ratio)
         else:
             img_height = args.img_size
diff --git a/Network.py b/Network.py
index 5110e10..bf7293e 100644
--- a/Network.py
+++ b/Network.py
@@ -187,7 +187,7 @@ def preprocess_image(image_path, load_dims=False, read_mode="color"):
         aspect_ratio = float(img_HEIGHT) / img_WIDTH
 
         img_width = args.img_size
-        if args.maintain_aspect_ratio:
+        if maintain_aspect_ratio:
             img_height = int(img_width * aspect_ratio)
         else:
             img_height = args.img_size
diff --git a/script_helper/Script/INetwork.py b/script_helper/Script/INetwork.py
index b51c432..833b36a 100644
--- a/script_helper/Script/INetwork.py
+++ b/script_helper/Script/INetwork.py
@@ -189,7 +189,7 @@ def preprocess_image(image_path, load_dims=False, read_mode="color"):
         aspect_ratio = float(img_HEIGHT) / img_WIDTH
 
         img_width = args.img_size
-        if args.maintain_aspect_ratio:
+        if maintain_aspect_ratio:
             img_height = int(img_width * aspect_ratio)
         else:
             img_height = args.img_size
diff --git a/script_helper/Script/Network.py b/script_helper/Script/Network.py
index 5110e10..bf7293e 100644
--- a/script_helper/Script/Network.py
+++ b/script_helper/Script/Network.py
@@ -187,7 +187,7 @@ def preprocess_image(image_path, load_dims=False, read_mode="color"):
         aspect_ratio = float(img_HEIGHT) / img_WIDTH
 
         img_width = args.img_size
-        if args.maintain_aspect_ratio:
+        if maintain_aspect_ratio:
             img_height = int(img_width * aspect_ratio)
         else:
             img_height = args.img_size