3 files changed, 3 insertions, 5 deletions

diff --git a/firmware/buflib.c b/firmware/buflib.c
index 0f8836b3b7..cb35290c03 100644
--- a/firmware/buflib.c
+++ b/firmware/buflib.c
@@ -894,7 +894,7 @@ bool
 buflib_shrink(struct buflib_context* ctx, int handle, void* new_start, size_t new_size)
 {
     char* oldstart = buflib_get_data(ctx, handle);
-    char* newstart = new_start;
+    char* newstart = new_start != NULL ? new_start : oldstart;
     char* newend = newstart + new_size;
 
     /* newstart must be higher and new_size not "negative" */
diff --git a/firmware/chunk_alloc.c b/firmware/chunk_alloc.c
index 6b80a475ab..85ad5d3489 100644
--- a/firmware/chunk_alloc.c
+++ b/firmware/chunk_alloc.c
@@ -172,12 +172,10 @@ static void finalize(struct chunk_alloc_header *hdr, struct chunk *chunk_array)
     int handle = chunk_array[idx].handle;
     struct buflib_context *ctx = hdr->context;
 
-    void* chunk_start = buflib_get_data(ctx, handle);
-
     hdr->chunk_bytes_total -= hdr->chunk_bytes_free;
     hdr->chunk_bytes_free = 0;
 
-    buflib_shrink(ctx, handle, chunk_start, hdr->chunk_bytes_total);
+    buflib_shrink(ctx, handle, NULL, hdr->chunk_bytes_total);
 
     logf("%s shrink hdr idx[%ld] offset[%ld]: new size: %ld",
     __func__, idx, chunk_array[idx].max_start_offset, hdr->chunk_bytes_total);
diff --git a/firmware/linuxboot.c b/firmware/linuxboot.c
index 74d87fbd64..6803bb466b 100644
--- a/firmware/linuxboot.c
+++ b/firmware/linuxboot.c
@@ -217,7 +217,7 @@ int uimage_load(struct uimage_header* uh, size_t* out_size,
     if(ret)
         goto err;
 
-    core_shrink(out_h, core_get_data(out_h), *out_size);
+    core_shrink(out_h, NULL, *out_size);
     ret = 0;
 
   err: