Set class on correct div.