If you only plan on doing light-moderate overclocking (or don't plan on overclocking) that would be fine, if you really want to push the chip you may want to make sure you get something with good VRMs. X370 boards don't automatically have better VRMs, but most of the high end boards with the best VRMs will be X370.
Edit: Oops, somehow missed that you already have the B350. In that case I wouldn't worry about getting a new mobo, that one should be fine.