diff --git a/scripts/imx/imx.c b/scripts/imx/imx.c
index 82ef97f..ca65389 100644
--- a/scripts/imx/imx.c
+++ b/scripts/imx/imx.c
@@ -221,6 +221,10 @@
 		if (!strcmp(socs[i].name, soc)) {
 			data->header_version = socs[i].header_version;
 			data->cpu_type = socs[i].cpu_type;
+
+			if (data->cpu_type == 35)
+				data->load_size += HEADER_LEN;
+
 			return 0;
 		}
 	}
@@ -230,9 +234,6 @@
 		fprintf(stderr, "%s ", socs[i].name);
 	fprintf(stderr, "\n");
 
-	if (data->cpu_type == 35)
-		data->load_size += HEADER_LEN;
-
 	return -EINVAL;
 }