Fix test for new AllocateAddress