Adding missing reference change