diff --git a/libretrogd/benches/bitmap.rs b/libretrogd/benches/bitmap.rs
index 2194052..c6dae24 100644
--- a/libretrogd/benches/bitmap.rs
+++ b/libretrogd/benches/bitmap.rs
@@ -1,7 +1,6 @@
 use criterion::{black_box, criterion_group, criterion_main, Criterion};
 
-use libretrogd::Bitmap;
-use libretrogd::Palette;
+use libretrogd::graphics::*;
 use libretrogd::{SCREEN_HEIGHT, SCREEN_WIDTH};
 
 pub fn criterion_benchmark(c: &mut Criterion) {
diff --git a/libretrogd/benches/blit.rs b/libretrogd/benches/blit.rs
index 63f74e0..c66477a 100644
--- a/libretrogd/benches/blit.rs
+++ b/libretrogd/benches/blit.rs
@@ -2,7 +2,8 @@ use std::path::Path;
 
 use criterion::{black_box, criterion_group, criterion_main, Criterion};
 
-use libretrogd::{Bitmap, BlitMethod, Rect};
+use libretrogd::graphics::*;
+use libretrogd::math::*;
 
 pub fn criterion_benchmark(c: &mut Criterion) {
     let mut framebuffer = Bitmap::new(320, 240).unwrap();