fix ml allocation size