tests pass, noob move

I forgot to make sure that my tests still pass after that UI change. Tests are all green now