Solving small issue with save button