Adopt to new display engine from edk2