Just small code improve, because I am maniac.