diff --git a/tests/integration-tests/tests/update/test_update.py b/tests/integration-tests/tests/update/test_update.py
index db97ed5673..fb6002666c 100644
--- a/tests/integration-tests/tests/update/test_update.py
+++ b/tests/integration-tests/tests/update/test_update.py
@@ -15,6 +15,7 @@
 import re
 import time
 from collections import defaultdict
+from datetime import datetime
 
 import boto3
 import pytest