Reduced time taken by graceful shutdown unit test.