Add zsmalloc back