added bug fix