fix mock
Merge pull request #51 from vieux/fix_mock

fix mock