Refactoring.
Minor fix.