Bugfix for setting "edit" button on first drawing