little code adjustment