Initial fix for undefined method save