diff --git a/transformations/tests/test_pool_allocator.py b/transformations/tests/test_pool_allocator.py
index 297bff942..85c7769ce 100644
--- a/transformations/tests/test_pool_allocator.py
+++ b/transformations/tests/test_pool_allocator.py
@@ -550,7 +550,7 @@ def test_pool_allocator_temporaries_kernel_sequence(frontend, block_dim, directi
     stack_size = f'max(c_sizeof(real(1, kind={kind_real}))*nlon + c_sizeof(real(1, kind={kind_real}))*nlon*nz + '
     stack_size += f'2*c_sizeof(int(1, kind={kind_int}))*nlon + c_sizeof(logical(true, kind={kind_log}))*nz,'
     stack_size += f'3*c_sizeof(real(1, kind={kind_real}))*nlon*nz + '
-    stack_size += f'c_sizeof(real(1,kind=jprb))*nlon)/c_sizeof(real(1, kind=jprb))'
+    stack_size += f'c_sizeof(real(1,kind={kind_real}))*nlon)/c_sizeof(real(1, kind=jprb))'
     check_stack_created_in_driver(driver, stack_size, calls[0], 2)
 
     # Has the data sharing been updated?