diff --git a/core/src/trezor/ui/__init__.py b/core/src/trezor/ui/__init__.py
index 672d1913e6..0498288db7 100644
--- a/core/src/trezor/ui/__init__.py
+++ b/core/src/trezor/ui/__init__.py
@@ -218,7 +218,7 @@ class Layout(Generic[T]):
         set_current_layout(self)
 
         # save context
-        self.context = context.get_context()
+        self.context = context.CURRENT_CONTEXT
 
         # attach a timer callback and paint self
         self._event(self.layout.attach_timer_fn, self._set_timer, transition_in)