6 files changed, 11 insertions, 1 deletions
diff --git a/examples/custom_shader/src/main.rs b/examples/custom_shader/src/main.rs
index e9b6776f..e7b07d78 100644
--- a/examples/custom_shader/src/main.rs
+++ b/examples/custom_shader/src/main.rs
@@ -9,6 +9,7 @@ use crate::pipeline::Pipeline;
 
 use iced::executor;
 use iced::time::Instant;
+use iced::widget::shader::wgpu;
 use iced::widget::{
     checkbox, column, container, row, shader, slider, text, vertical_space,
 };
diff --git a/examples/custom_shader/src/pipeline.rs b/examples/custom_shader/src/pipeline.rs
index 44ad17a2..9343e5e0 100644
--- a/examples/custom_shader/src/pipeline.rs
+++ b/examples/custom_shader/src/pipeline.rs
@@ -1,8 +1,10 @@
 use crate::primitive;
 use crate::primitive::cube;
 use crate::primitive::{Buffer, Uniforms};
+use crate::wgpu;
+use crate::wgpu::util::DeviceExt;
+
 use iced::{Rectangle, Size};
-use wgpu::util::DeviceExt;
 
 const SKY_TEXTURE_SIZE: u32 = 128;
 
diff --git a/examples/custom_shader/src/primitive.rs b/examples/custom_shader/src/primitive.rs
index 520ceb8e..f5862ab3 100644
--- a/examples/custom_shader/src/primitive.rs
+++ b/examples/custom_shader/src/primitive.rs
@@ -9,6 +9,7 @@ pub use cube::Cube;
 pub use uniforms::Uniforms;
 pub use vertex::Vertex;
 
+use crate::wgpu;
 use crate::Camera;
 use crate::Pipeline;
 
diff --git a/examples/custom_shader/src/primitive/buffer.rs b/examples/custom_shader/src/primitive/buffer.rs
index 377ce1bb..ef4c41c9 100644
--- a/examples/custom_shader/src/primitive/buffer.rs
+++ b/examples/custom_shader/src/primitive/buffer.rs
@@ -1,3 +1,5 @@
+use crate::wgpu;
+
 // A custom buffer container for dynamic resizing.
 pub struct Buffer {
     pub raw: wgpu::Buffer,
diff --git a/examples/custom_shader/src/primitive/cube.rs b/examples/custom_shader/src/primitive/cube.rs
index c23f2132..7ece685d 100644
--- a/examples/custom_shader/src/primitive/cube.rs
+++ b/examples/custom_shader/src/primitive/cube.rs
@@ -1,4 +1,6 @@
 use crate::primitive::Vertex;
+use crate::wgpu;
+
 use glam::{vec2, vec3, Vec3};
 use rand::{thread_rng, Rng};
 
diff --git a/examples/custom_shader/src/primitive/vertex.rs b/examples/custom_shader/src/primitive/vertex.rs
index 6d17aa0f..e64cd926 100644
--- a/examples/custom_shader/src/primitive/vertex.rs
+++ b/examples/custom_shader/src/primitive/vertex.rs
@@ -1,3 +1,5 @@
+use crate::wgpu;
+
 #[derive(Debug, Copy, Clone, bytemuck::Pod, bytemuck::Zeroable)]
 #[repr(C)]
 pub struct Vertex {