diff --git a/apax/nodes/bal.py b/apax/nodes/bal.py
index b40ccd0b..c69d52be 100644
--- a/apax/nodes/bal.py
+++ b/apax/nodes/bal.py
@@ -95,4 +95,4 @@ def _get_plot(self, atoms_lst: list[ase.Atoms], indices: list[int]):
 
         ax.plot(indices, energies[indices], "x", color="red")
 
-        fig.savefig(self.img_selection, bbox_inches="tight")
\ No newline at end of file
+        fig.savefig(self.img_selection, bbox_inches="tight")
diff --git a/apax/nodes/md.py b/apax/nodes/md.py
index 2ca833d1..e35f380d 100644
--- a/apax/nodes/md.py
+++ b/apax/nodes/md.py
@@ -104,4 +104,4 @@ def file_handle(filename):
             format_handler=functools.partial(
                 znh5md.FormatHandler, file_handle=file_handle
             ),
-        ).get_atoms_list()
\ No newline at end of file
+        ).get_atoms_list()