diff --git a/frontend/ui/rendertext.lua b/frontend/ui/rendertext.lua
index 47f8c8bb5..6a0ca34cc 100644
--- a/frontend/ui/rendertext.lua
+++ b/frontend/ui/rendertext.lua
@@ -6,10 +6,10 @@ local Font = require("ui/font")
 local Cache = require("cache")
 local CacheItem = require("cacheitem")
 local BlitBuffer = require("ffi/blitbuffer")
+local Device = require("device")
 local logger = require("logger")
 
-local CanvasContext = require("document/canvascontext")
-if CanvasContext.should_restrict_JIT then
+if Device.should_restrict_JIT then
     require("jit").off(true, true)
 end
 
diff --git a/frontend/ui/widget/container/inputcontainer.lua b/frontend/ui/widget/container/inputcontainer.lua
index fdbb74cea..dd5cbb0d4 100644
--- a/frontend/ui/widget/container/inputcontainer.lua
+++ b/frontend/ui/widget/container/inputcontainer.lua
@@ -30,11 +30,11 @@ local Geom = require("ui/geometry")
 local GestureRange = require("ui/gesturerange")
 local UIManager = require("ui/uimanager")
 local WidgetContainer = require("ui/widget/container/widgetcontainer")
+local Device = require("device")
+local Screen = Device.screen
 local _ = require("gettext")
-local Screen = require("device").screen
 
-local CanvasContext = require("document/canvascontext")
-if CanvasContext.should_restrict_JIT then
+if Device.should_restrict_JIT then
     require("jit").off(true, true)
 end
 
diff --git a/frontend/ui/widget/container/widgetcontainer.lua b/frontend/ui/widget/container/widgetcontainer.lua
index 257a796a9..c285c6cac 100644
--- a/frontend/ui/widget/container/widgetcontainer.lua
+++ b/frontend/ui/widget/container/widgetcontainer.lua
@@ -15,9 +15,9 @@ It handles event propagation and painting (with different alignments) for its ch
 
 local Geom = require("ui/geometry")
 local Widget = require("ui/widget/widget")
+local Device = require("device")
 
-local CanvasContext = require("document/canvascontext")
-if CanvasContext.should_restrict_JIT then
+if Device.should_restrict_JIT then
     require("jit").off(true, true)
 end