fixed tests for UI changes.