Removed mock in favor of flexmock