Also you can just use `wraps` like the test above to avoid having to do this at all.