Updated to match refactored methods in the mockery-core.