Fix/ui tests