Fix for bug.