diff --git a/src/eth_plugin_ui.c b/src/eth_plugin_ui.c
index 0d1fa5a..fa126d6 100644
--- a/src/eth_plugin_ui.c
+++ b/src/eth_plugin_ui.c
@@ -79,7 +79,7 @@ void display_next_plugin_item(bool entering) {
                 // Reset multi page layout to the first page
                 G_ux.layout_paging.current = 0;
 #ifdef TARGET_NANOS
-                ux_layout_paging_redisplay(G_ux.stack_count - 1);
+                ux_layout_paging_redisplay_by_addr(G_ux.stack_count - 1);
 #else
                 ux_layout_bnnn_paging_redisplay(0);
 #endif
diff --git a/src/ui_callbacks.h b/src/ui_callbacks.h
index 5fc97a9..26702bc 100644
--- a/src/ui_callbacks.h
+++ b/src/ui_callbacks.h
@@ -1,4 +1,5 @@
 #include "shared_context.h"
+#include "ux.h"
 
 unsigned int io_seproxyhal_touch_settings(const bagl_element_t *e);
 unsigned int io_seproxyhal_touch_exit(const bagl_element_t *e);
diff --git a/src/ui_flow.h b/src/ui_flow.h
index 4e8f321..927f5e2 100644
--- a/src/ui_flow.h
+++ b/src/ui_flow.h
@@ -1,6 +1,7 @@
 #include "shared_context.h"
 
 #include "os_io_seproxyhal.h"
+#include "ux.h"
 
 extern const ux_flow_step_t* const ux_idle_flow[];