diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index 1cc44acf0b5965f983c38a722822d2750849458b..9f7aebdb1be93789a20dd92afdfee5c2d825f0eb 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -17,8 +17,8 @@ jobs:
       matrix:
         os: [ubuntu-latest, macOS-latest]
         python: ['3.8']
-        torch: ['1.8.0']
-        torchvision: ['0.9.0']
+        torch: ['1.8.1']
+        torchvision: ['0.9.1']
     runs-on: ${{ matrix.os }}
 
     steps:
diff --git a/tests/test_models.py b/tests/test_models.py
index 63a95fa5884f2b5e731885897168c0901b0017c3..029ae0dd40ac272f7ae56a81b5e7fc0529631d81 100644
--- a/tests/test_models.py
+++ b/tests/test_models.py
@@ -73,7 +73,7 @@ def test_model_forward(model_name, batch_size):
     model = create_model(model_name, pretrained=False)
     model.eval()
 
-    input_size = _get_input_size(model, TARGET_FWD_SIZE)
+    input_size = _get_input_size(model=model, target=TARGET_FWD_SIZE)
     if max(input_size) > MAX_FWD_SIZE:
         pytest.skip("Fixed input size model > limit.")
     inputs = torch.randn((batch_size, *input_size))
@@ -221,7 +221,7 @@ def test_model_forward_features(model_name, batch_size):
     expected_channels = model.feature_info.channels()
     assert len(expected_channels) >= 4  # all models here should have at least 4 feature levels by default, some 5 or 6
 
-    input_size = _get_input_size(model, TARGET_FFEAT_SIZE)
+    input_size = _get_input_size(model=model, target=TARGET_FFEAT_SIZE)
     if max(input_size) > MAX_FFEAT_SIZE:
         pytest.skip("Fixed input size model > limit.")