Add bug fix