Fixing a minor bug.