Fix tests for Mockery 0.9