Fix graceful shutdown for update-blocks.