fixed minor bug