Minor optimization for the simple DIET.