Bug fix

fixed minor bug