1 files changed, 5 insertions, 7 deletions
diff --git a/libkmstest/kmstest.h b/libkmstest/kmstest.h
index 0564ab5..aed40fe 100644
--- a/libkmstest/kmstest.h
+++ b/libkmstest/kmstest.h
@@ -1,17 +1,15 @@
 #pragma once
 
 #include "color.h"
+#include "framebuffer.h"
 
 namespace kms
 {
-class MappedBuffer;
-class DumbFramebuffer;
+class IMappedFramebuffer;
 
-void draw_color_bar(kms::DumbFramebuffer& buf, int old_xpos, int xpos, int width);
+void draw_color_bar(IMappedFramebuffer& buf, int old_xpos, int xpos, int width);
 
-void draw_test_pattern(MappedBuffer& fb);
-void draw_test_pattern(DumbFramebuffer &fb);
+void draw_test_pattern(IMappedFramebuffer &fb);
 
-void draw_rect(MappedBuffer &fb, uint32_t x, uint32_t y, uint32_t w, uint32_t h, RGB color);
-void draw_rect(DumbFramebuffer &fb, uint32_t x, uint32_t y, uint32_t w, uint32_t h, RGB color);
+void draw_rect(IMappedFramebuffer &fb, uint32_t x, uint32_t y, uint32_t w, uint32_t h, RGB color);
 }