Small fix on save button