Update to correct a naming issue