Add updated mock to 1.0.1