diff --git a/src/graphnet/datasets/nubench_datasets.py b/src/graphnet/datasets/nubench_datasets.py
index 0f49f2707..3465ac5ce 100644
--- a/src/graphnet/datasets/nubench_datasets.py
+++ b/src/graphnet/datasets/nubench_datasets.py
@@ -20,7 +20,6 @@
 )
 from graphnet.training.labels import Direction, Track
 
-
 FEATURES_NUBENCH = [
     "sensor_pos_x",
     "sensor_pos_y",