diff --git a/src/differences/attgt/attgt.py b/src/differences/attgt/attgt.py
index 392e87e..b2361ae 100644
--- a/src/differences/attgt/attgt.py
+++ b/src/differences/attgt/attgt.py
@@ -336,7 +336,7 @@ def _create_result_dict(
                 data=(
                     self.data
                     if not self._dropped_nas_flag
-                    else self.data[split_sample_by].loc[
+                    else self.data.loc[
                         lambda x: x.index.isin(self._data_matrix.index)
                     ]
                 ),