diff --git a/src/qusi/single_dense_layer_model.py b/src/qusi/single_dense_layer_model.py
index fed8ffe9..f5a070e2 100644
--- a/src/qusi/single_dense_layer_model.py
+++ b/src/qusi/single_dense_layer_model.py
@@ -1,10 +1,9 @@
 from __future__ import annotations
 
-from typing_extensions import Self
-
 import torch
 from torch import Tensor
 from torch.nn import Linear, Module, Sigmoid
+from typing_extensions import Self
 
 
 class SingleDenseLayerBinaryClassificationModel(Module):
diff --git a/src/qusi/train_session.py b/src/qusi/train_session.py
index 405f3bb2..28f41ed3 100644
--- a/src/qusi/train_session.py
+++ b/src/qusi/train_session.py
@@ -37,7 +37,7 @@ def train_session(
         process_rank=0,
         project=logging_configuration.wandb_project,
         entity=logging_configuration.wandb_entity,
-        settings=wandb.Settings(start_method="fork"),
+        settings=wandb.Settings(start_method="thread"),
     )
     sessions_directory = Path("sessions")
     sessions_directory.mkdir(exist_ok=True)