diff --git a/src/ShaderNode/ShaderGraph.cpp b/src/ShaderNode/ShaderGraph.cpp
index e2cbe6a2b..c0b33c4ae 100644
--- a/src/ShaderNode/ShaderGraph.cpp
+++ b/src/ShaderNode/ShaderGraph.cpp
@@ -464,7 +464,7 @@ nzsl::Ast::ModulePtr ShaderGraph::ToModule() const
 	for (const auto& structInfo : m_structs)
 	{
 		nzsl::Ast::StructDescription structDesc;
-		structDesc.layout = nzsl::StructLayout::Std140;
+		structDesc.layout = nzsl::Ast::MemoryLayout::Std140;
 		structDesc.name = structInfo.name;
 
 		for (const auto& memberInfo : structInfo.members)
diff --git a/src/ShaderNode/Widgets/MainWindow.cpp b/src/ShaderNode/Widgets/MainWindow.cpp
index 194da1a6b..22a4b4b47 100644
--- a/src/ShaderNode/Widgets/MainWindow.cpp
+++ b/src/ShaderNode/Widgets/MainWindow.cpp
@@ -192,7 +192,7 @@ void MainWindow::OnCompile()
 
 		Nz::File file(fileName.toStdString(), Nz::OpenMode::WriteOnly);
 		nzsl::Serializer serializer;
-		nzsl::Ast::SerializeShader(serializer, shaderModule);
+		nzsl::Ast::SerializeShader(serializer, *shaderModule);
 
 		const std::vector<std::uint8_t>& data = serializer.GetData();
 		file.Write(data.data(), data.size());