Update for Allocation API changes.

Change-Id: I46bbeab13df570fd14a21c4a2790adc9665dca26
1 file changed