Refactor a tiny bit