3 files changed, 3 insertions, 2 deletions
diff --git a/py/functest.py b/py/functest.py
index 21a543a..24a9a85 100755
--- a/py/functest.py
+++ b/py/functest.py
@@ -8,7 +8,7 @@ conn = card.get_first_connected_connector()
 
 mode = conn.get_default_mode()
 
-fb = pykms.Framebuffer(card, mode.hdisplay, mode.vdisplay, "XR24");
+fb = pykms.DumbFramebuffer(card, mode.hdisplay, mode.vdisplay, "XR24");
 pykms.draw_test_pattern(fb);
 
 crtc = conn.get_current_crtc()
diff --git a/py/pykms.i b/py/pykms.i
index d14067e..c71aced 100644
--- a/py/pykms.i
+++ b/py/pykms.i
@@ -17,6 +17,7 @@ using namespace kms;
 %include "card.h"
 %include "property.h"
 %include "framebuffer.h"
+%include "dumbframebuffer.h"
 %include "plane.h"
 %include "connector.h"
 %include "encoder.h"
diff --git a/py/test.py b/py/test.py
index a104ba5..a4a4f8d 100755
--- a/py/test.py
+++ b/py/test.py
@@ -9,7 +9,7 @@ conn = card.get_first_connected_connector()
 
 mode = conn.get_default_mode()
 
-fb = pykms.Framebuffer(card, mode.hdisplay, mode.vdisplay, "XR24");
+fb = pykms.DumbFramebuffer(card, mode.hdisplay, mode.vdisplay, "XR24");
 pykms.draw_test_pattern(fb);
 
 crtc = conn.get_current_crtc()