diff --git a/maestro/trainer/models/qwen2_vl/loaders.py b/maestro/trainer/models/qwen2_vl/loaders.py
index b88d39a..f84766c 100644
--- a/maestro/trainer/models/qwen2_vl/loaders.py
+++ b/maestro/trainer/models/qwen2_vl/loaders.py
@@ -1,4 +1,3 @@
-
 from transformers.pipelines.base import Dataset
 
 from maestro.trainer.common.data_loaders.datasets import JSONLDataset
@@ -17,18 +16,7 @@ def __getitem__(self, idx):
         suffix = data["suffix"]
         return {
             "messages": [
-                {
-                    "role": "user",
-                    "content": [
-                        {"type": "image", "image": image},
-                        {"type": "text", "text": prefix}
-                    ]
-                },
-                {
-                    "role": "assistant",
-                    "content": [
-                        {"type": "text", "text": suffix}
-                    ]
-                }
+                {"role": "user", "content": [{"type": "image", "image": image}, {"type": "text", "text": prefix}]},
+                {"role": "assistant", "content": [{"type": "text", "text": suffix}]},
             ]
         }