fixed a small bug
saving works