Get ui tests back working