--- a/src/Makefile
+++ b/src/Makefile
@@ -5,4 +5,6 @@
 
 obj-$(CONFIG_BE2NET) += be2net.o
 
+ccflags-y += -DCONFIG_BE2NET_BE2 -DCONFIG_BE2NET_BE3
+
 be2net-y :=  be_main.o be_cmds.o be_ethtool.o be_roce.o