The way I read it, you didn't replace or change anything except for the bearings, and there wasn't a problem before you took the headset apart.
Is that right?
Is it possible that you used replacement bearings that are smaller than the original bearings, or you installed the wrong number of bearings?