From 80332c0d5a53acd57743ee36bdeb80a22f002025 Mon Sep 17 00:00:00 2001
From: LilDojd <yawner@protonmail.com>
Date: Sun, 17 Nov 2024 02:39:38 +0400
Subject: [PATCH] Fix tests

---
 alchemiscale/storage/statestore.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/alchemiscale/storage/statestore.py b/alchemiscale/storage/statestore.py
index 23f97482..366c8a54 100644
--- a/alchemiscale/storage/statestore.py
+++ b/alchemiscale/storage/statestore.py
@@ -5,7 +5,7 @@
 """
 
 import abc
-from datetime import UTC, datetime
+from datetime import datetime
 from contextlib import contextmanager
 import json
 from functools import lru_cache
@@ -1795,7 +1795,7 @@ def task_count(task_dict: dict):
                 tx.run(
                     CLAIM_QUERY,
                     tasks_list=[str(task) for task in tasks if task is not None],
-                    datetimestr=datetime.now(UTC).isoformat(),
+                    datetimestr=str(datetime.utcnow().isoformat()),
                     compute_service_id=str(compute_service_id),
                 )