diff --git a/medcat-v2/tests/components/linking/test_context_based_linker.py b/medcat-v2/tests/components/linking/test_context_based_linker.py
index 119cc7278..a295f0f67 100644
--- a/medcat-v2/tests/components/linking/test_context_based_linker.py
+++ b/medcat-v2/tests/components/linking/test_context_based_linker.py
@@ -1,4 +1,5 @@
 from medcat.components.linking import context_based_linker
+from medcat.components.types import _DEFAULT_LINKING as DEF_LINKING
 from medcat.components import types
 from medcat.config import Config
 from medcat.vocab import Vocab
@@ -35,7 +36,7 @@ def weighted_average_function(self, nr: int) -> float:
 
 
 class LinkingInitTests(ComponentInitTests, unittest.TestCase):
-    expected_def_components = 3
+    expected_def_components = len(DEF_LINKING)
     comp_type = types.CoreComponentType.linking
     default_cls = context_based_linker.Linker
     default_creator = context_based_linker.Linker.create_new_component
diff --git a/medcat-v2/tests/components/ner/test_vocab_based_ner.py b/medcat-v2/tests/components/ner/test_vocab_based_ner.py
index 5a971b8aa..5c0ca03fe 100644
--- a/medcat-v2/tests/components/ner/test_vocab_based_ner.py
+++ b/medcat-v2/tests/components/ner/test_vocab_based_ner.py
@@ -26,7 +26,7 @@ def __init__(self, config: Config):
 
 
 class NerInitTests(ComponentInitTests, unittest.TestCase):
-    expected_def_components = 3
+    expected_def_components = len(types._DEFAULT_NER)
     comp_type = types.CoreComponentType.ner
     default_cls = vocab_based_ner.NER
     default_creator = vocab_based_ner.NER.create_new_component