small code fixing