Update to make specs pass.