diff --git a/tests/benchmark/fixtures/FFTFixture.h b/tests/benchmark/fixtures/FFTFixture.h
index c9c4e3a..53897b1 100644
--- a/tests/benchmark/fixtures/FFTFixture.h
+++ b/tests/benchmark/fixtures/FFTFixture.h
@@ -36,8 +36,8 @@
 {
 namespace benchmark
 {
-template <typename TensorType, typename Function, typename Accessor>
-class FFT1DFixture : public framework::Fixture
+template <typename TensorType, typename Function, typename FFTInfo, typename Accessor>
+class FFTFixture : public framework::Fixture
 {
 public:
     template <typename...>
@@ -48,7 +48,7 @@
         dst = create_tensor<TensorType>(shape, data_type, 2);
 
         // Create and configure function
-        fft_func.configure(&src, &dst, FFT1DInfo());
+        fft_func.configure(&src, &dst, FFTInfo());
 
         // Allocate tensors
         src.allocator()->allocate();