diff --git a/sonic-pde-tests/sonic_pde_saiut/Makefile b/sonic-pde-tests/sonic_pde_saiut/Makefile
index 054318b..3b533e9 100644
--- a/sonic-pde-tests/sonic_pde_saiut/Makefile
+++ b/sonic-pde-tests/sonic_pde_saiut/Makefile
@@ -1,8 +1,10 @@
 SAI_INC := /usr/include/sai
 SAI_LIB := /usr/lib
 
+
+PYTHON_CFLAGS := $(shell python3-config --includes)
 CC      := gcc
-CFLAGS  := -fPIC -I/usr/include/python3.11 -I$(SAI_INC) -I$(SAI_INC)/experimental
+CFLAGS  := -fPIC $(PYTHON_CFLAGS) -I$(SAI_INC) -I$(SAI_INC)/experimental
 
 LDFLAGS := -L$(SAI_LIB) -lsai
 
@@ -11,17 +13,22 @@ all: _saiut.so
 setup:
 	rm -rf .inc
 	mkdir -p .inc
-	ln -s $(SAI_INC)/saitypes.h .inc/
-	ln -s $(SAI_INC)/saistatus.h .inc/
-	ln -s $(SAI_INC)/saiport.h .inc/
+	ln -sf $(SAI_INC)/saitypes.h .inc/
+	ln -sf $(SAI_INC)/saistatus.h .inc/
+	ln -sf $(SAI_INC)/saiport.h .inc/
 
 saiut_wrap.c: setup saiut.i
 	swig -python saiut.i
 
 saiut_wrap.o: saiut_wrap.c
+	$(CC) $(CFLAGS) -c $< -o $@
+
+saiut.o: saiut.c
+	$(CC) $(CFLAGS) -c $< -o $@
 
 _saiut.so: saiut.o saiut_wrap.o
-	gcc -shared $^ $(LDFLAGS) -o $@
+	$(CC) -shared $^ $(LDFLAGS) -o $@
 
 clean:
 	rm -rf *.so *.o *_wrap.c saiut.py saiut.pyc .inc/
+