Make sure we get the object for the right mock class