diff --git a/hidl/fingerprint/UdfpsExtension.cpp b/hidl/fingerprint/UdfpsExtension.cpp
index 494bcaf..7daf9b8 100644
--- a/hidl/fingerprint/UdfpsExtension.cpp
+++ b/hidl/fingerprint/UdfpsExtension.cpp
@@ -15,7 +15,12 @@
  */
 
 #include <compositionengine/UdfpsExtension.h>
+
+#if __has_include(<display/drm/sde_drm.h>)
+#include <display/drm/sde_drm.h>
+#elif __has_include(<drm/sde_drm.h>)
 #include <drm/sde_drm.h>
+#endif
 
 uint32_t getUdfpsZOrder(uint32_t z, bool touched) {
 #ifdef FOD_PRESSED_LAYER_ZORDER