diff --git a/drivers/staging/android/ion/ion_system_secure_heap.c b/drivers/staging/android/ion/ion_system_secure_heap.c
index a7cb38d81173..a3f85418dad5 100644
--- a/drivers/staging/android/ion/ion_system_secure_heap.c
+++ b/drivers/staging/android/ion/ion_system_secure_heap.c
@@ -66,6 +66,7 @@ static void ion_system_secure_heap_free(struct ion_buffer *buffer)
 	struct ion_system_secure_heap *secure_heap =
 		to_system_secure_heap(heap);
 
+	buffer->heap = secure_heap->sys_heap;
 	secure_heap->sys_heap->ops->free(buffer);
 }