Updating unit tests to new css names