Set proper handler after reconnecting