Fix allocator on entity