corrections before UI demo