diff --git a/src/restart_conditions/unhealthy_nodes.js b/src/restart_conditions/unhealthy_nodes.js
index 1ed8959..4fe94f1 100644
--- a/src/restart_conditions/unhealthy_nodes.js
+++ b/src/restart_conditions/unhealthy_nodes.js
@@ -1,5 +1,6 @@
 import axios from "axios"
 import { CHECK_NODE_HEALTHY_LIMIT, DYNAMO_RESTART_TYPES, LAYERS, RESTART_REASONS } from "../utils/types.js"
+import { sleep } from "../shared/shared.js"
 
 const _getUnhealthyNodes = async (
   clusterInfo