Fixed bug with save code.