diff --git a/src/cosmosdb-preview/azext_cosmosdb_preview/tests/latest/test_cosmosdb-InAccountRestore_scenario.py b/src/cosmosdb-preview/azext_cosmosdb_preview/tests/latest/test_cosmosdb-InAccountRestore_scenario.py
index 40f22c68584..5690d4f1cee 100644
--- a/src/cosmosdb-preview/azext_cosmosdb_preview/tests/latest/test_cosmosdb-InAccountRestore_scenario.py
+++ b/src/cosmosdb-preview/azext_cosmosdb_preview/tests/latest/test_cosmosdb-InAccountRestore_scenario.py
@@ -8,12 +8,15 @@
 
 from azure.cli.testsdk.scenario_tests import AllowLargeResponse
 from azure.cli.testsdk import (ScenarioTest, ResourceGroupPreparer)
-from datetime import datetime
+from datetime import datetime, timedelta, timezone
 import datetime
+from knack.log import get_logger
 
 TEST_DIR = os.path.abspath(os.path.join(os.path.abspath(__file__), '..'))
 
 
+logger = get_logger(__name__)
+
 class Cosmosdb_previewInAccountRestoreScenarioTest(ScenarioTest):
 
 #sql