Update to match documentation