Fixing a minor issue