should fix ui tests