Fixed bug with UI code which caused certain pieces to not render