diff --git a/keras_lmu/tests/test_benchmarks.py b/keras_lmu/tests/test_benchmarks.py
index d58de82e..dffb1832 100644
--- a/keras_lmu/tests/test_benchmarks.py
+++ b/keras_lmu/tests/test_benchmarks.py
@@ -38,7 +38,7 @@ def on_predict_batch_end(self, batch, logs=None):
 @pytest.mark.skipif(not tf_gpu_installed, reason="Very slow on CPU")
 @pytest.mark.parametrize(
     "mode, min_time, max_time",
-    [("rnn", 0.1, 0.2), ("fft", 0.1, 0.2), ("raw", 0.05, 0.15)],
+    [("rnn", 0.1, 0.2), ("fft", 0.05, 0.15), ("raw", 0.05, 0.15)],
 )
 def test_performance(mode, min_time, max_time):
     # performance is based on Azure NC6 VM