Fix tests for Mockery 0.9. Closes #1