Fix failing test after CSS refactor