diff --git a/src/lib/memrange.c b/src/lib/memrange.c
index 96d75242..79a1b0e 100644
--- a/src/lib/memrange.c
+++ b/src/lib/memrange.c
@@ -47,7 +47,7 @@
 		range_entry_unlink(&ranges->free_list, r);
 		return r;
 	}
-	if (ENV_RAMSTAGE)
+	if (ENV_PAYLOAD_LOADER)
 		return malloc(sizeof(struct range_entry));
 	return NULL;
 }