Fixing the issue when I missed adding the save()