3 files changed, 11 insertions, 2 deletions
diff --git a/src/tests/test_demo_effects.cc b/src/tests/test_demo_effects.cc
index d0163c2..0d2b09a 100644
--- a/src/tests/test_demo_effects.cc
+++ b/src/tests/test_demo_effects.cc
@@ -197,6 +197,9 @@ static void test_effect_type_classification() {
 int main() {
   fprintf(stdout, "=== Demo Effects Tests ===\n");
 
+  extern void InitShaderComposer();
+  InitShaderComposer();
+
   test_post_process_effects();
   test_scene_effects();
   test_effect_type_classification();
diff --git a/src/tests/test_effect_base.cc b/src/tests/test_effect_base.cc
index e280e05..612e9da 100644
--- a/src/tests/test_effect_base.cc
+++ b/src/tests/test_effect_base.cc
@@ -249,6 +249,9 @@ static void test_pixel_helpers() {
 int main() {
   fprintf(stdout, "=== Effect Base Tests ===\n");
 
+  extern void InitShaderComposer();
+  InitShaderComposer();
+
   test_webgpu_fixture();
   test_offscreen_render_target();
   test_effect_construction();
diff --git a/src/tests/test_shader_compilation.cc b/src/tests/test_shader_compilation.cc
index e2c0adc..a322e8a 100644
--- a/src/tests/test_shader_compilation.cc
+++ b/src/tests/test_shader_compilation.cc
@@ -115,16 +115,19 @@ static bool test_shader_compilation(const char* name, const char* shader_code) {
     return true; // Not a failure, just skipped
   }
 
+  // Compose shader to resolve #include directives
+  std::string composed_shader = ShaderComposer::Get().Compose({}, shader_code);
+
 #if defined(DEMO_CROSS_COMPILE_WIN32)
   WGPUShaderModuleWGSLDescriptor wgsl_desc = {};
   wgsl_desc.chain.sType = WGPUSType_ShaderModuleWGSLDescriptor;
-  wgsl_desc.code = shader_code;
+  wgsl_desc.code = composed_shader.c_str();
   WGPUShaderModuleDescriptor shader_desc = {};
   shader_desc.nextInChain = (const WGPUChainedStruct*)&wgsl_desc.chain;
 #else
   WGPUShaderSourceWGSL wgsl_desc = {};
   wgsl_desc.chain.sType = WGPUSType_ShaderSourceWGSL;
-  wgsl_desc.code = str_view(shader_code);
+  wgsl_desc.code = str_view(composed_shader.c_str());
   WGPUShaderModuleDescriptor shader_desc = {};
   shader_desc.nextInChain = (const WGPUChainedStruct*)&wgsl_desc.chain;
 #endif