1 files changed, 7 insertions, 15 deletions

diff --git a/src/tests/gpu/test_shader_composer.cc b/src/tests/gpu/test_shader_composer.cc
index 4002863..6e0b707 100644
--- a/src/tests/gpu/test_shader_composer.cc
+++ b/src/tests/gpu/test_shader_composer.cc
@@ -48,29 +48,21 @@ void test_asset_composition() {
 
   size_t snippet_a_size;
   const char* snippet_a_code =
-      (const char*)GetAsset(AssetId::ASSET_SHADER_SNIPPET_A, &snippet_a_size);
+      (const char*)GetAsset(AssetId::ASSET_TEST_WGSL, &snippet_a_size);
   assert(snippet_a_code != nullptr);
-  sc.RegisterSnippet("SNIPPET_A", std::string(snippet_a_code, snippet_a_size));
-
-  size_t snippet_b_size;
-  const char* snippet_b_code =
-      (const char*)GetAsset(AssetId::ASSET_SHADER_SNIPPET_B, &snippet_b_size);
-  sc.RegisterSnippet("SNIPPET_B", std::string(snippet_b_code, snippet_b_size));
+  sc.RegisterSnippet("TEST_WGSL", std::string(snippet_a_code, snippet_a_size));
 
   std::string main_code =
-      "fn main() -> f32 { return snippet_a() + snippet_b(); }";
-  std::string result = sc.Compose({"SNIPPET_A", "SNIPPET_B"}, main_code);
+      "fn main() -> f32 { return test_wgsl(); }";
+  std::string result = sc.Compose({"TEST_WGSL"}, main_code);
 
-  assert(result.find("fn snippet_a()") != std::string::npos);
-  assert(result.find("fn snippet_b()") != std::string::npos);
+  assert(result.find("fn test_wgsl()") != std::string::npos);
   assert(result.find("fn main()") != std::string::npos);
 
-  size_t pos_a = result.find("snippet_a");
-  size_t pos_b = result.find("snippet_b");
+  size_t pos_a = result.find("test_wgsl");
   size_t pos_main = result.find("main");
 
-  assert(pos_a < pos_b);
-  assert(pos_b < pos_main);
+  assert(pos_a < pos_main);
 
   std::cout << "Asset-based composition logic verified." << std::endl;
 }