PartialViewResults are not getting Models from TempData imported

I changed ImportModelStateFromTempDataAttribute to check if the result is ViewResultBase instead of ViewResult so it includes PartialViewResults like the comment implies.