Small bit og refactoring.