fixed minor bug with saving