Fixing bug with click save on modals