fixed up ui testing