diff --git a/ohmpi/hardware_components/mb_2024_0_2.py b/ohmpi/hardware_components/mb_2024_0_2.py
index 52113afecfd627a8d3ee6b2dd2a73e4220779af7..a4b65db3bc621943ef287c5c416806e0ca304f15 100644
--- a/ohmpi/hardware_components/mb_2024_0_2.py
+++ b/ohmpi/hardware_components/mb_2024_0_2.py
@@ -66,7 +66,7 @@ def _ads_1115_gain_auto(channel):  # Make it a class method ?
 
 class Tx(Tx_mb_2023):
     def __init__(self, **kwargs):
-        if kwargs['model'] == os.path.basename(__file__).rstrip('.py'):
+        if 'model' not in kwargs.keys():
             for key in SPECS['tx'].keys():
                 kwargs = enforce_specs(kwargs, SPECS['tx'], key)
             subclass_init = False
@@ -93,7 +93,7 @@ class Tx(Tx_mb_2023):
 
 class Rx(Rx_mb_2023):
     def __init__(self, **kwargs):
-        if kwargs['model'] == os.path.basename(__file__).rstrip('.py'):
+        if 'model' not in kwargs.keys():
             for key in SPECS['rx'].keys():
                 kwargs = enforce_specs(kwargs, SPECS['rx'], key)
             subclass_init = False