Fixing a little bug.