fix: ensure the ATG is properly restored after disconnection to CSMS