Fix broken setAddButton in StoryView