diff --git a/netbox_pcie.c b/netbox_pcie.c
index 934368e..d982b78 100644
--- a/netbox_pcie.c
+++ b/netbox_pcie.c
@@ -36,7 +36,6 @@
 #include <linux/kern_levels.h>
 #include <linux/version.h>
 
-#include <nbsw.h>
 #include <misc/nrsw-global.h>
 #include <linux/board_descriptor.h>
 
@@ -806,7 +805,6 @@ static int nm_pcie_probe(struct platform_device *pdev)
 	pcie_slot_count = 0;
 	hwver = hwrel = -1;
 
-#ifdef NBSW_TARGET_netbolt_arm
 	if (NBHW == 14) {
 		for (i = 0; i < 6; i++) {
 			ps = pcie_add_slot(i, PCIE_MAIN);
@@ -850,8 +848,7 @@ static int nm_pcie_probe(struct platform_device *pdev)
 			if (ret)
 				goto error;
 		}
-#elif defined(NBSW_TARGET_netbird_arm)
-	if (NBHW == 16) {
+	} else if (NBHW == 16) {
 		ps = pcie_add_slot(0, PCIE_MAIN);
 		if (IS_ERR(ps)) {
 			ret = PTR_ERR(ps);
@@ -1116,7 +1113,6 @@ static int nm_pcie_probe(struct platform_device *pdev)
 		ret = pcie_init_slot(ps);
 		if (ret)
 			goto error;
-#endif
 	} else {
 		printk("Unknwon NBHW: %d\n", NBHW);
 		ret = -EINVAL;