Modify method save to pass tests.