diff --git a/turbo_alignment/dataset/chat/chat.py b/turbo_alignment/dataset/chat/chat.py
index c3a7656..2cd23e1 100755
--- a/turbo_alignment/dataset/chat/chat.py
+++ b/turbo_alignment/dataset/chat/chat.py
@@ -376,8 +376,7 @@ def get_slice(self, start: int, end: int) -> Self:
 
         new_instance.records = self.records[start:end]
         new_instance.original_records_map = {
-            record['id']: self.get_original_record_by_id(record['id'])
-            for record in dataset_records
+            record['id']: self.get_original_record_by_id(record['id']) for record in dataset_records
         }
 
         return new_instance