Fix the save w/ callback test