Fixed ui rendering bug

Closes #106