diff --git a/mopidy_touchscreen/screen_objects.py b/mopidy_touchscreen/screen_objects.py index 94c2e2d..f2ceed0 100644 --- a/mopidy_touchscreen/screen_objects.py +++ b/mopidy_touchscreen/screen_objects.py @@ -26,8 +26,10 @@ class ScreenObjectsManager(): self.text_objects = {} def set_object(self, key, add_object): - if self.text_objects[key] is not None: + try: self.dirty_area.append(self.text_objects[key].rect_in_pos) + except KeyError: + pass self.text_objects[key] = add_object self.dirty_area.append(add_object.rect_in_pos)