diff --git a/src/render.rs b/src/render.rs
index 3a75cd4..03674f6 100644
--- a/src/render.rs
+++ b/src/render.rs
@@ -1,5 +1,5 @@
 use std::io;
-use rgb::RGB8;
+use rgb::Rgb;
 
 use crate::geometry::*;
 use crate::task::Task;
@@ -9,7 +9,7 @@ use std::marker::PhantomData;
 use std::fmt::Debug;
 
 pub trait Shader: Send + Debug {
-    fn draw(&self, surface_coords: &VirtualCoordinates, frame: usize) -> RGB8;
+    fn draw(&self, surface_coords: &VirtualCoordinates, frame: usize) -> Rgb<u8>;
 }
 
 pub trait Surfaces<T: Surface>: Debug {