coorection bug

correction