Fix for minor bug.