Small edits on TEAllocator