Little code improvement