How to solve the Elon Musk problem in one bill or less

https://www.law.cornell.edu/uscode/text/18/208

This is 18 USC 208, which prohibits people working in or with the executive branch of the government from profiting from this affiliation. It contains enough loopholes so Musk can continue to run his companies while he's being president. All that needs done is to get rid of the loopholes in it so one of three things happens: Musk completely divests himself from all the companies he is in charge of - and by "completely divests" I mean he's gotta sell his stock too; Musk decides to not be president nor advise Trump in any way; or the government ends its SpaceX, Starlink and Tesla contracts.