diff --git a/src/great_ai/great_ai/persistence/parallel_tinydb_driver.py b/src/great_ai/great_ai/persistence/parallel_tinydb_driver.py
index 1243ab0..61b0761 100644
--- a/src/great_ai/great_ai/persistence/parallel_tinydb_driver.py
+++ b/src/great_ai/great_ai/persistence/parallel_tinydb_driver.py
@@ -82,8 +82,8 @@ class ParallelTinyDbDriver(TracingDatabaseDriver):
 
         if sort_by:
             df.sort_values(
-                [col["column_id"] for col in sort_by],
-                ascending=[col["direction"] == "asc" for col in sort_by],
+                [col.column_id for col in sort_by],
+                ascending=[col.direction == "asc" for col in sort_by],
                 inplace=True,
             )
 
diff --git a/src/great_ai/great_ai/persistence/tracing_database_driver.py b/src/great_ai/great_ai/persistence/tracing_database_driver.py
index 51a7be3..e26a8ae 100644
--- a/src/great_ai/great_ai/persistence/tracing_database_driver.py
+++ b/src/great_ai/great_ai/persistence/tracing_database_driver.py
@@ -1,4 +1,4 @@
-from abc import ABC, abstractclassmethod, abstractmethod
+from abc import ABC, abstractmethod
 from datetime import datetime
 from pathlib import Path
 from typing import List, Optional, Sequence, Tuple, Union
@@ -18,7 +18,7 @@ class TracingDatabaseDriver(ABC):
     ) -> None:
         cls.configure_credentials(**ConfigFile(secrets_path))
 
-    @abstractclassmethod
+    @classmethod
     def configure_credentials(
         cls,
     ) -> None: