small error fix

small error fix