Update references after move