fix save method