It certainly can be added to the first stage by moving the entire resistive feedback network from the second opamp to the first, but just connect R4 to the sense node instead of the output of the opamp.
However, this doesn't really accomplish quite the same function. Now, the current range of the dummy load will be increased by factor of 2, and measuring the real current draw must be done by measuring across the sense resistor itself.