I think the OP made it very clear what he was worried about and it was/is a fantasy.

BATF&E may have opined whatever they want but I'd be surprised if there is a single case that has ever been prosecuted or in brought of a person that, with no other crime involved, owned a gun, voluntarily when to rehab for marijuana, got out successfully and then was arrested.