It's up to you but if it were my tax return, I would go with the 1099-B the broker gives me and the IRS, exactly.
But wait until March to be sure you are not going to get a corrected 1099-B.
