Fixed UI message error on running tests