Restore callbacks in new Session UI