Note this applies only to citizens and lawful permanent residents. I’m not sure if there’s an underlying constitutional limitation or if it’s arbitrary, but (IANAL) it seems arbitrary as the cited section has no such limitation.
I welcome any progress towards decriminalization and legalization, but I think it’s important always to be aware when and how the progress is limited.