diff --git a/tests/attributes_test.py b/tests/attributes_test.py
index 2271690..621bbf9 100644
--- a/tests/attributes_test.py
+++ b/tests/attributes_test.py
@@ -1,4 +1,3 @@
-import typing
 import pytest
 import torch
 from grassmann_tensor import GrassmannTensor
@@ -16,7 +15,7 @@
     ((), (), torch.randn([])),
     ((False, False, True), ((2, 2), (1, 3), (4, 0)), torch.randn([4, 4, 4])),
 ])
-def x(request: typing.Any) -> Initialization:
+def x(request: pytest.FixtureRequest) -> Initialization:
     return request.param
 
 
diff --git a/tests/update_mask_test.py b/tests/update_mask_test.py
index 42123ba..7118fc0 100644
--- a/tests/update_mask_test.py
+++ b/tests/update_mask_test.py
@@ -1,4 +1,3 @@
-import typing
 import pytest
 import torch
 from grassmann_tensor import GrassmannTensor
@@ -18,7 +17,7 @@
         torch.randn([2, 4, 2], dtype=torch.float32),
     ),
 ])
-def x(request: typing.Any) -> Initialization:
+def x(request: pytest.FixtureRequest) -> Initialization:
     return request.param