vis_model_transfer tests are broken

[bhazelton]

These broke after #84 was merged but we didn't catch it in that PR because these tests aren't run in GitHub actions.