small code structure fix