fix errors in UI