adding bug fix