Fix reference after renaming