Fix bug where the visualisation edit dialog would show on the published page