tiny code refinement