1 files changed, 2 insertions, 2 deletions
diff --git a/tests/test_devices_tosa_advice_generation.py b/tests/test_devices_tosa_advice_generation.py
index 018ba57..1b97c8b 100644
--- a/tests/test_devices_tosa_advice_generation.py
+++ b/tests/test_devices_tosa_advice_generation.py
@@ -1,7 +1,7 @@
 # SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
 # SPDX-License-Identifier: Apache-2.0
 """Tests for advice generation."""
-from typing import List
+from __future__ import annotations
 
 import pytest
 
@@ -40,7 +40,7 @@ def test_tosa_advice_producer(
     tmpdir: str,
     input_data: DataItem,
     advice_category: AdviceCategory,
-    expected_advice: List[Advice],
+    expected_advice: list[Advice],
 ) -> None:
     """Test TOSA advice producer."""
     producer = TOSAAdviceProducer()