Updated reference to renamed class