Fixing #52.

Mocks will always trick you into believing you're done when you're not.
Evil little things