Add source bug fix