diff --git a/apps/CodegenHeatEquationWithMaterial.cpp b/apps/CodegenHeatEquationWithMaterial.cpp
index b93f83ad320920d88eb688b1f620b5fd4efe647f..324791ca5770a6dbde5d3fbae1a6615d1eaaf056 100644
--- a/apps/CodegenHeatEquationWithMaterial.cpp
+++ b/apps/CodegenHeatEquationWithMaterial.cpp
@@ -21,7 +21,7 @@
 
 #include "timeloop/SweepTimeloop.h"
 
-#include "gen/HeatEquationWithMaterial/HeatEquationKernelWithMaterial.h"
+#include "gen/HeatEquationKernelWithMaterial.hpp"
 
 namespace walberla
 {
diff --git a/apps/HeatEquationKernelWithMaterial.py b/apps/HeatEquationKernelWithMaterial.py
index 5a9b95570b00eb0add6bd7cdf514aba600ac2c46..1b7e1cfeabdb06a3589a306e2eb1d814aab0f52f 100644
--- a/apps/HeatEquationKernelWithMaterial.py
+++ b/apps/HeatEquationKernelWithMaterial.py
@@ -9,6 +9,7 @@ from sfg_walberla import Sweep
 from pymatlib.data.alloys import Ti6Al4V
 from pymatlib.data.alloys.SS316L import SS316L
 from pymatlib.core.assignment_converter import assignment_converter
+from pymatlib.core.interpolators import DoubleLookupArrayContainer
 
 with SourceFileGenerator() as sfg:
     data_type = "float64"  # if ctx.double_accuracy else "float32"
@@ -26,6 +27,9 @@ with SourceFileGenerator() as sfg:
 
     # mat = Ti6Al4V.create_Ti6Al4V(u.center())
     mat = SS316L.create_SS316L(u.center())
+    # arr_container = DoubleLookupArrayContainer("SS316L", mat.temperature_array, mat.energy_density_array)
+    arr_container = DoubleLookupArrayContainer.from_material("SS316L", mat)
+    sfg.generate(arr_container)
 
     # Convert assignments to pystencils format
     print("Print statements")