fixed minor bug

fixed minor bug