fix ui tests