Fixed unnoticed broken tests after setting things up in the ui (sorry!)