How are you removing the ED from the ZigBee Network? Is there any specific reason that you want the short address?
Remember when the end device sends the leave command this address had been erased from the table.
Please look at the API uint16 ZPS_u16AplZdoLookupAddr(uint64 u64ExtAddr)
Document ZigBee 3.0 Stack User Guide