Refs #151 - Fix mocking code.