diff --git a/src/gretel_trainer/benchmark/__init__.py b/src/gretel_trainer/benchmark/__init__.py
index 9f76eced..0d8520aa 100644
--- a/src/gretel_trainer/benchmark/__init__.py
+++ b/src/gretel_trainer/benchmark/__init__.py
@@ -15,5 +15,6 @@
     GretelGPTX,
     GretelLSTM,
     GretelModel,
+    GretelNavigatorFT,
 )
 from gretel_trainer.benchmark.log import set_log_level
diff --git a/src/gretel_trainer/benchmark/entrypoints.py b/src/gretel_trainer/benchmark/entrypoints.py
index 022cf5d7..fb2fcc5d 100644
--- a/src/gretel_trainer/benchmark/entrypoints.py
+++ b/src/gretel_trainer/benchmark/entrypoints.py
@@ -92,7 +92,7 @@ def _entrypoint(
 def _verify_client_config(session: ClientConfig):
     try:
         current_user_email = session.email
-        logger.info(f"Using gretel client configured with {current_user_email}!r")
+        logger.info(f"Using gretel client configured with {current_user_email!r}")
     except Exception as e:
         raise BenchmarkException(
             "Invalid gretel client configuration, please make sure to configure the "
diff --git a/src/gretel_trainer/benchmark/session.py b/src/gretel_trainer/benchmark/session.py
index 569cea03..e24a9c3a 100644
--- a/src/gretel_trainer/benchmark/session.py
+++ b/src/gretel_trainer/benchmark/session.py
@@ -30,6 +30,8 @@
     "PCS": "principal_component_stability",
     "FDS": "field_distribution_stability",
     "PPL": "privacy_protection_level",
+    "AIA": "attribute_inference_attack_score",
+    "DPS": "data_privacy_score",
 }
 
 FutureKeyT = Union[str, RunKey]