We used the solution here:
These tires work a lot better than the original. The only issue we ran into is that the wider tires will sometimes rub against the robot frame. Our fix for this was to slightly modify the 3D printed wheels such that the connector that slides onto the motor shaft is a little bit shorter so that the wheel sits a little closer to the motor (and therefore a bit farther away from the robot frame)