tried to fix save bug