Slight correction to previous PR