Fix broken allocator usage