bug fix in code

The code has been fixed