Fix failing test_save case