Little refactor.