You are correct if we are counting the legalization for "medical" purposes.  Booze is another big profit item for the government and costs society a lot.  No way to control human vices so it is much better and profitable to tax them.