Adding code to replace lost properties in UI