small code correction