diff --git a/cmd/source.c b/cmd/source.c
index d724d63eb7..45e9794b2f 100644
--- a/cmd/source.c
+++ b/cmd/source.c
@@ -84,7 +84,7 @@ source (ulong addr, const char *fit_uname)
 		 * past the zero-terminated sequence of image lengths to get
 		 * to the actual image data
 		 */
-		while (*data++);
+		while (*data++ != IMAGE_PARAM_INVAL);
 		break;
 #endif
 #if defined(CONFIG_FIT)