Fix build errors in UMP driver