diff --git a/dataquality/__init__.py b/dataquality/__init__.py
index 40eebaf92..44afa265b 100644
--- a/dataquality/__init__.py
+++ b/dataquality/__init__.py
@@ -30,7 +30,7 @@
         dataquality.get_insights()
 """
 
-__version__ = "v0.8.42"
+__version__ = "v0.8.43"
 
 import sys
 from typing import Any, List, Optional
diff --git a/dataquality/integrations/setfit.py b/dataquality/integrations/setfit.py
index 248b64a33..450789276 100644
--- a/dataquality/integrations/setfit.py
+++ b/dataquality/integrations/setfit.py
@@ -246,7 +246,7 @@ def __call__(self, *args: Any, **kwds: Any) -> Any:
                 # 🔭🌕 Galileo logging
                 dq.log_model_outputs(
                     ids=batch["id"],
-                    logits=dq_store["output"],
+                    probs=dq_store["output"],
                     embs=dq_store["input_args"][0],
                     split=split,
                     epoch=0,
@@ -362,7 +362,7 @@ def dq_evaluate(
             # 🔭🌕 Galileo logging
             dq.log_model_outputs(
                 ids=batch[id_col],
-                logits=dq_store["output"],
+                probs=dq_store["output"],
                 embs=dq_store["input_args"][0],
                 split=split,
                 epoch=0,