malloc_simple: Correct the alignment logic in memalign_simple()

This should use the align parameter, not bytes. Natural alignment is one
use case but should not be the only one supported by this function.

Signed-off-by: Simon Glass <sjg@chromium.org>
Reviewed-by: Tom Rini <trini@konsulko.com>
diff --git a/common/malloc_simple.c b/common/malloc_simple.c
index 9811ab6..134e059 100644
--- a/common/malloc_simple.c
+++ b/common/malloc_simple.c
@@ -31,7 +31,7 @@
 	ulong addr, new_ptr;
 	void *ptr;
 
-	addr = ALIGN(gd->malloc_base + gd->malloc_ptr, bytes);
+	addr = ALIGN(gd->malloc_base + gd->malloc_ptr, align);
 	new_ptr = addr + bytes;
 	if (new_ptr > gd->malloc_limit)
 		return NULL;