Graceful shutdown for UpdateModule