Simplify ms_allocator and as_allocator design