Fix for edit section button