Two mini refactors