Fix save test