diff --git a/projects/common/src/client/java/dan200/computercraft/client/gui/widgets/DynamicImageButton.java b/projects/common/src/client/java/dan200/computercraft/client/gui/widgets/DynamicImageButton.java
index 6c7324688..5f67a86ac 100644
--- a/projects/common/src/client/java/dan200/computercraft/client/gui/widgets/DynamicImageButton.java
+++ b/projects/common/src/client/java/dan200/computercraft/client/gui/widgets/DynamicImageButton.java
@@ -68,14 +68,11 @@ public void renderWidget(PoseStack stack, int mouseX, int mouseY, float partialT
         RenderSystem.enableDepthTest();
     }
 
-    @Override
-    public Component getMessage() {
-        return message.get().message;
-    }
-
     @Override
     public void render(PoseStack stack, int mouseX, int mouseY, float partialTicks) {
-        setTooltip(message.get().tooltip());
+        var message = this.message.get();
+        setMessage(message.message());
+        setTooltip(message.tooltip());
         super.render(stack, mouseX, mouseY, partialTicks);
     }