#2711 Fixed failing test in dofkern_test.py

stfc · Oct 9, 2024 · 70150d3 · 70150d3
1 parent 2704972
commit 70150d3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/psyclone/tests/domain/lfric/dofkern_test.py b/src/psyclone/tests/domain/lfric/dofkern_test.py
@@ -291,7 +291,7 @@ def test_function_space_bare_undf():
                            TEST_API,
                            idx=0, dist_mem=False)
     schedule = invoke.schedule
-    kernel = schedule.walk(LFRicKern)
+    kernel = schedule.walk(LFRicKern)[0]
     test_fs = DynFunctionSpaces(kernel)
 
     assert 'undf' in test_fs._var_list