change allocation model to reflect API change