Fixed error due to renaming

Sorry...