diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 8aeca94..420aaee 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -57,4 +57,4 @@ jobs:
           if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
       - name: Test with pytest
         run: |
-          pytest -n 0 --durations 0 -v -m "not slow" -k ${{ matrix.pytest-file }}
+          pytest -n 0 --durations 0 -v -m "not slow" ${{ matrix.pytest-file }}
diff --git a/jetnet/datasets/normalisations.py b/jetnet/datasets/normalisations.py
index e81c8a2..f6141f4 100644
--- a/jetnet/datasets/normalisations.py
+++ b/jetnet/datasets/normalisations.py
@@ -1,6 +1,7 @@
 """
 Suite of common ways to normalise data.
 """
+
 from __future__ import annotations
 
 from abc import ABC, abstractmethod
diff --git a/jetnet/datasets/utils.py b/jetnet/datasets/utils.py
index 3d7dc89..4be8e73 100644
--- a/jetnet/datasets/utils.py
+++ b/jetnet/datasets/utils.py
@@ -1,6 +1,7 @@
 """
 Utility methods for datasets.
 """
+
 from __future__ import annotations
 
 import hashlib