Add the logout callback to prevent exception

Add the logout callback to prevent exception #232