Little code optimization