Fix test bug for saving

Now master will pass :)