In general I'd like to see some more tests here, but I imagine poking the UI is probably difficult, so I think it's probably okay here.