Fix slab allocator, add small test