Keep it simple... Chances are, the shower head is where flow is restricted. Just use a shower head that allows more flow. You may test it by removing the shower head from the hose and try if the heater shuts down or not. If not, the shower head is the culprit.