diff --git a/dataset.py b/dataset.py
index c717595..4a4e87c 100644
--- a/dataset.py
+++ b/dataset.py
@@ -34,7 +34,7 @@ def __getitem__(self, index):
 
         # setting system information
         if self.system_format is not None:
-            system = data["system"].strip() if "system" in data.keys() else self.system
+            system = data["system"].strip() if "system" in data.keys() else self.system_format
 
             if system is not None:
                 system_text = self.system_format.format(content=system)