Fix broken test. Due to CSS class rename