updated tests for ui