Fixing a minor bug