Fix to update source visualization when attached to a volume