little code fixing