Fixed a UI bug I introduced on the last commit.