Fix test for save