Corrected a small error in refactor