Previous UI update broke the test suite, fixed