ui: Fix error class in tests