Mini refac.
Refac.