fixing a minor bug