Correct save bug