diff --git a/pyproject.toml b/pyproject.toml
index baec4b1..203cd25 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -10,7 +10,7 @@ include = '\.pyi?$'
 
 [project]
 name = "aiia"
-version = "0.1.2"
+version = "0.1.3"
 description = "AIIA Deep Learning Model Implementation"
 readme = "README.md"
 authors = [
diff --git a/setup.cfg b/setup.cfg
index ff97663..4b75cda 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -1,6 +1,6 @@
 [metadata]
 name = aiia
-version = 0.1.2
+version = 0.1.3
 author = Falko Habel
 author_email = falko.habel@gmx.de
 description = AIIA deep learning model implementation
diff --git a/src/aiia/__init__.py b/src/aiia/__init__.py
index 65857b4..f4edbbf 100644
--- a/src/aiia/__init__.py
+++ b/src/aiia/__init__.py
@@ -4,4 +4,4 @@ from .data.DataLoader import DataLoader
 from .pretrain.pretrainer import Pretrainer, ProjectionHead
 
 
-__version__ = "0.1.2"
+__version__ = "0.1.3"
diff --git a/src/aiia/pretrain/pretrainer.py b/src/aiia/pretrain/pretrainer.py
index 1b128c0..42ba4b8 100644
--- a/src/aiia/pretrain/pretrainer.py
+++ b/src/aiia/pretrain/pretrainer.py
@@ -6,7 +6,7 @@ from tqdm import tqdm
 from ..model.Model import AIIA
 from ..model.config import AIIAConfig
 from ..data.DataLoader import AIIADataLoader
-
+import os
 
 class ProjectionHead(nn.Module):
     def __init__(self, hidden_size):
@@ -189,7 +189,8 @@ class Pretrainer:
                 self.model.save(output_path)
                 print("Best model saved!")
 
-        self.save_losses('losses.csv')
+        losses_path = os.path.join(os.path.dirname(output_path), 'losses.csv')
+        self.save_losses(losses_path)
 
     def _validate(self, val_loader, criterion_denoise, criterion_rotate):
         """Perform validation and return average validation loss."""