fix ui bug with boxes