Two failed water heater inspections drip leg

The way I have installed drip legs on water heaters is short nipple out of control valve. Tee. 6" drip leg and cap. Gas flex into top of the tee. Gas valve before flex. Was told today on both heaters. Drip leg needs to be before gas shut off valve. Flex after drip leg. Is this new, or have I been doing it wrong for twenty years? One inspection was in county, the other in city. Two separate jurisdictions.

Via RSSMix.com Mix ID 8247361 http://www.rssmix.com/