Fix tests because of new name