Uh, small refactor.