Update references to fix build.