Error from refactor fixed.