diff --git a/src/writer/hlsl/generator_impl_test.cc b/src/writer/hlsl/generator_impl_test.cc
index 39b39dc..4055bb9 100644
--- a/src/writer/hlsl/generator_impl_test.cc
+++ b/src/writer/hlsl/generator_impl_test.cc
@@ -28,18 +28,17 @@
 
 using HlslGeneratorImplTest = TestHelper;
 
-TEST_F(HlslGeneratorImplTest, DISABLED_Generate) {
+TEST_F(HlslGeneratorImplTest, Generate) {
   ast::type::VoidType void_type;
   mod()->AddFunction(std::make_unique<ast::Function>(
       "my_func", ast::VariableList{}, &void_type));
   mod()->AddEntryPoint(std::make_unique<ast::EntryPoint>(
-      ast::PipelineStage::kFragment, "my_func", ""));
+      ast::PipelineStage::kFragment, "", "my_func"));
 
   ASSERT_TRUE(gen().Generate(out())) << gen().error();
-  EXPECT_EQ(result(), R"(#import <metal_lib>
-
-void my_func() {
+  EXPECT_EQ(result(), R"(void my_func() {
 }
+
 )");
 }
 
diff --git a/src/writer/msl/generator_impl_test.cc b/src/writer/msl/generator_impl_test.cc
index f57bf52..af5600e 100644
--- a/src/writer/msl/generator_impl_test.cc
+++ b/src/writer/msl/generator_impl_test.cc
@@ -45,21 +45,22 @@
 
 using MslGeneratorImplTest = testing::Test;
 
-TEST_F(MslGeneratorImplTest, DISABLED_Generate) {
+TEST_F(MslGeneratorImplTest, Generate) {
   ast::type::VoidType void_type;
   ast::Module m;
   m.AddFunction(std::make_unique<ast::Function>("my_func", ast::VariableList{},
                                                 &void_type));
   m.AddEntryPoint(std::make_unique<ast::EntryPoint>(
-      ast::PipelineStage::kCompute, "my_func", ""));
+      ast::PipelineStage::kCompute, "", "my_func"));
 
   GeneratorImpl g(&m);
 
   ASSERT_TRUE(g.Generate()) << g.error();
-  EXPECT_EQ(g.result(), R"(#import <metal_lib>
+  EXPECT_EQ(g.result(), R"(#include <metal_stdlib>
 
-compute void my_func() {
+kernel void my_func() {
 }
+
 )");
 }