Little fix on save function