The tone control is post distortion, you have to tap off the input with a buffer of some description in to the 'Dry' input.
It's just a simple place to add it, you could also re-jigger the output op-amp... wherever you put it, you need a buffer from the input and some mixing resistors.