diff --git a/assets/shaders/line_material.wgsl b/assets/shaders/line_material.wgsl
index 02f278f60f058b..e47ffe6e16acb3 100644
--- a/assets/shaders/line_material.wgsl
+++ b/assets/shaders/line_material.wgsl
@@ -1,11 +1,13 @@
 struct LineMaterial {
-    color: vec4<f32>;
+    color: vec4<f32>,
 };
 
-[[group(1), binding(0)]]
+@group(1) @binding(0)
 var<uniform> material: LineMaterial;
 
-[[stage(fragment)]]
-fn fragment() -> [[location(0)]] vec4<f32> {
+@fragment
+fn fragment(
+    #import bevy_pbr::mesh_vertex_output
+) -> @location(0) vec4<f32> {
     return material.color;
-}
\ No newline at end of file
+}