Updating: fixing a minor bug