Update allocator.c

Additional fixes