From 32b2cded53cb5984a2c660d2d46e863ab5dc6566 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Szymon=20=C5=81opaciuk?= <szymon@lopaciuk.eu>
Date: Tue, 5 Mar 2024 10:00:06 +0100
Subject: [PATCH] fix in test

---
 tests/test_prebuild_kernels.py | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/tests/test_prebuild_kernels.py b/tests/test_prebuild_kernels.py
index 12269e641..2fdfc31ba 100644
--- a/tests/test_prebuild_kernels.py
+++ b/tests/test_prebuild_kernels.py
@@ -85,23 +85,23 @@ def test_prebuild_kernels(mocker, tmp_path, temp_context_default_func, capsys):
 
 def test_per_element_prebuild_kernels(mocker, tmp_path, temp_context_default_func):
     # Set up the temporary kernels directory
-    kernel_definitions = {
-        "test_module": {
+    kernel_definitions = [
+        ("test_module", {
             "config": {},
             "classes": [
                 xt.Drift,
                 xt.Cavity,
                 xt.XYShift,
             ]
-        },
-        "test_module_rand": {
+        }),
+        ("test_module_rand", {
             "config": {},
             "classes": [],
             "extra_classes": [
                 xt.RandomNormal,
             ]
-        },
-    }
+        }),
+    ]
 
     patch_defs = 'xtrack.prebuilt_kernels.kernel_definitions.kernel_definitions'
     mocker.patch(patch_defs, kernel_definitions)