diff --git a/src/graphnet/models/graphs/graph_definition.py b/src/graphnet/models/graphs/graph_definition.py
index 042e54c58..aa73693ef 100644
--- a/src/graphnet/models/graphs/graph_definition.py
+++ b/src/graphnet/models/graphs/graph_definition.py
@@ -306,7 +306,9 @@ def _add_features_individually(
                 graph[feature] = graph.x[:, index].detach()
             else:
                 self.warning_once(
-                    """Cannot assign graph['x']. This field is reserved for node features. Please rename your input feature."""
+                    """Cannot assign graph['x']. This field is reserved for
+                      node features. Please rename your input feature."""
+                )  # noqa
 
         return graph
 
diff --git a/tests/utilities/test_model_config.py b/tests/utilities/test_model_config.py
index 68ebd5c2a..8979f0255 100644
--- a/tests/utilities/test_model_config.py
+++ b/tests/utilities/test_model_config.py
@@ -18,7 +18,7 @@
 
 
 def test_simple_model_config(path: str = "/tmp/simple_model.yml") -> None:
-    """Test saving, loading, and reconstructing using a simple model."""
+    """Test saving, loading, and reconstructing simple model."""
     # Construct single Model
     model = DynEdge(
         nb_inputs=9,