Refactore a bit.