Refactor minor issue.