diff --git a/tests/cmake_integration/kernels.py b/tests/cmake_integration/kernels.py
index d780b8f04669b5b33ee5210e91cc03a60a762723..901bdb02befbd4024b4921ad6c921b8188638a8a 100644
--- a/tests/cmake_integration/kernels.py
+++ b/tests/cmake_integration/kernels.py
@@ -1,20 +1,20 @@
+# type: ignore
+
 import sympy as sp
-import numpy as np
 
-from pystencils.session import *
+from pystencils import fields, kernel
 
 from pystencilssfg import SourceFileGenerator
-from pystencilssfg.source_concepts.cpp import std_mdspan
 
 
 with SourceFileGenerator() as sfg:
-    src, dst = ps.fields("src, dst(1) : double[2D]")
+    src, dst = fields("src, dst(1) : double[2D]")
 
     h = sp.Symbol('h')
 
-    @ps.kernel
+    @kernel
     def poisson_jacobi():
-        dst[0,0] @= (src[1, 0] + src[-1, 0] + src[0, 1] + src[0, -1]) / 4
+        dst[0, 0] @= (src[1, 0] + src[-1, 0] + src[0, 1] + src[0, -1]) / 4
 
     poisson_kernel = sfg.kernels.create(poisson_jacobi)
 
diff --git a/tests/cmake_integration/more_kernels.py b/tests/cmake_integration/more_kernels.py
index 2d97a94c0f99427e345447ef7650728637242928..75b798fb7b30431613531edee033df1512d23212 100644
--- a/tests/cmake_integration/more_kernels.py
+++ b/tests/cmake_integration/more_kernels.py
@@ -1,20 +1,20 @@
+# type: ignore
+
 import sympy as sp
-import numpy as np
 
-from pystencils.session import *
+from pystencils import fields, kernel, Field
 
 from pystencilssfg import SourceFileGenerator
-from pystencilssfg.source_concepts.cpp import std_mdspan
 
 
 with SourceFileGenerator() as sfg:
-    src = ps.fields("src: double[2D]")
+    src: Field = fields("src: double[2D]")
 
     h = sp.Symbol('h')
 
-    @ps.kernel
+    @kernel
     def poisson_gs():
-        src[0,0] @= (src[1, 0] + src[-1, 0] + src[0, 1] + src[0, -1]) / 4
+        src[0, 0] @= (src[1, 0] + src[-1, 0] + src[0, 1] + src[0, -1]) / 4
 
     poisson_kernel = sfg.kernels.create(poisson_gs)