Minor performance fix