fix ui tests
Merge pull request #579 from ascott1/ui-test-fix

fix ui tests