diff --git a/toot/tui/timeline.py b/toot/tui/timeline.py
index ab8c78bc..3d72eb73 100644
--- a/toot/tui/timeline.py
+++ b/toot/tui/timeline.py
@@ -338,9 +338,9 @@ def update_status_image(self, status, path, placeholder_index):
                 pass
         if img:
             try:
-                status.placeholders[placeholder_index]._set_original_widget(
-                    graphics_widget(img, image_format=self.tui.options.image_format, corner_radius=10,
-                                    colors=self.tui.options.colors))
+                status.placeholders[placeholder_index].original_widget = graphics_widget(
+                    img, image_format=self.tui.options.image_format, corner_radius=10, colors=self.tui.options.colors
+                )
 
             except IndexError:
                 # ignore IndexErrors.
diff --git a/toot/tui/widgets.py b/toot/tui/widgets.py
index db7bf9ee..0d20cc4f 100644
--- a/toot/tui/widgets.py
+++ b/toot/tui/widgets.py
@@ -58,7 +58,7 @@ def __init__(self, *args, **kwargs):
 
     def get_state(self):
         """Return the state of the checkbox."""
-        return self.button._state
+        return self.button.get_state()
 
 
 class RadioButton(urwid.AttrWrap):