a few more tweaks to AllocationModel