Updated test
- Mock object wasn't correctly implementing method