Actual refactor