diff --git a/apps/plugins/test_mem.c b/apps/plugins/test_mem.c
index 80995c9bfb..da89e13340 100644
--- a/apps/plugins/test_mem.c
+++ b/apps/plugins/test_mem.c
@@ -69,8 +69,7 @@ enum plugin_status plugin_start(const void* parameter)
                               KB_PER_SEC(delta));
         rb->screens[0]->update();
 
-        int button = rb->button_get(false);
-        switch (button)
+        switch (rb->get_action(CONTEXT_STD, TIMEOUT_NOBLOCK))
         {
 #ifdef HAVE_ADJUSTABLE_CPU_FREQ
             case ACTION_STD_PREV: