Update for new source