diff --git a/components/frekvens_panel/display.py b/components/frekvens_panel/display.py
index 7afec82..65f686b 100644
--- a/components/frekvens_panel/display.py
+++ b/components/frekvens_panel/display.py
@@ -49,6 +49,6 @@ async def to_code(config):
 
     if CONF_LAMBDA in config:
         lambda_ = await cg.process_lambda(
-            config[CONF_LAMBDA], [(display.DisplayBufferRef, "it")], return_type=cg.void
+            config[CONF_LAMBDA], [(display.DisplayRef, "it")], return_type=cg.void
         )
         cg.add(var.set_writer(lambda_))