minor change

bug fix