diff --git a/tests/restore.py b/tests/restore.py
index bbdadeb23..733f9caba 100644
--- a/tests/restore.py
+++ b/tests/restore.py
@@ -4,7 +4,11 @@
 import subprocess
 import sys
 from time import sleep
-from datetime import datetime, timedelta, timezone
+from datetime import datetime, timedelta
+try:
+    from datetime import timezone
+except ImportError:  # py2compat
+    from pytz import timezone
 import hashlib
 import shutil
 import json