Updated the code to reflect the name change