modified tests to handle the new save method