Updated code to be compatible with changes in DPU-API module