a little code improvement