tiny refactor nitpick