Kristen Stewart battled COVID on set