Minor source fix