Update references in script