bug fix

fixed bug issue 6