Fix the rest of the py scripts

1 files changed, 1 insertions, 2 deletions

diff --git a/py/db.py b/py/db.py
index c51ca70..315f993 100755
--- a/py/db.py
+++ b/py/db.py
@@ -33,13 +33,12 @@ class FlipHandler(pykms.PageFlipHandlerBase):
 
         if card.has_atomic:
             ctx = pykms.AtomicReq(card)
-            ctx.add(crtc, "FB_ID", fb.id)
+            ctx.add(crtc.primary_plane, "FB_ID", fb.id)
             ctx.commit(self)
         else:
             crtc.page_flip(fb, self)
 
 
-
 card = pykms.Card()
 conn = card.get_first_connected_connector()
 mode = conn.get_default_mode()