Fix save_version method