[Demo] Update for UI testing.