fix typehint

1 files changed, 2 insertions, 2 deletions

diff --git a/tests/test_nvd_dataset_builder.py b/tests/test_nvd_dataset_builder.py
index 0e71764c..bc69cc3c 100644
--- a/tests/test_nvd_dataset_builder.py
+++ b/tests/test_nvd_dataset_builder.py
@@ -1,6 +1,6 @@
 from datetime import date, datetime, timedelta
 from importlib import resources
-from typing import Any
+from typing import Any, Type
 
 import pytest
 
@@ -50,7 +50,7 @@ def test_cpe_match_download_from_seccerts():
         ({"timestamp": datetime.fromtimestamp(0).isoformat(), "match_strings": {}}, CpeMatchNvdDatasetBuilder),
     ],
 )
-def test_build_dataset(default_dataset: Any, builder_class: NvdDatasetBuilder):
+def test_build_dataset(default_dataset: Any, builder_class: Type[NvdDatasetBuilder]):
     def get_update_timestamp_from_dataset(dset) -> datetime:
         if isinstance(dset, (CPEDataset, CVEDataset)):
             return dset.last_update_timestamp