diff --git a/test/test_data.h b/test/test_data.h
index a04be83..fea3b6c 100644
--- a/test/test_data.h
+++ b/test/test_data.h
@@ -274,6 +274,9 @@ static const BitMasks g_vaSurfaceMemTypes = {
     VA_SURFACE_ATTRIB_MEM_TYPE_USER_PTR, VA_SURFACE_ATTRIB_MEM_TYPE_KERNEL_DRM,
     VA_SURFACE_ATTRIB_MEM_TYPE_DRM_PRIME,
     VA_SURFACE_ATTRIB_MEM_TYPE_DRM_PRIME_2,
+#if VA_CHECK_VERSION(1, 21, 0)
+    VA_SURFACE_ATTRIB_MEM_TYPE_DRM_PRIME_3,
+#endif
 };
 
 static const std::map<VASurfaceAttribType, const BitMasks&>
diff --git a/test/test_va_api_createsurfaces.cpp b/test/test_va_api_createsurfaces.cpp
index 2376b49..156afc1 100644
--- a/test/test_va_api_createsurfaces.cpp
+++ b/test/test_va_api_createsurfaces.cpp
@@ -88,6 +88,9 @@ class VAAPISurfaceFixture
 
         const uint32_t drmMemMask = VA_SURFACE_ATTRIB_MEM_TYPE_KERNEL_DRM
                                     | VA_SURFACE_ATTRIB_MEM_TYPE_DRM_PRIME
+#if VA_CHECK_VERSION(1, 21, 0)
+                                    | VA_SURFACE_ATTRIB_MEM_TYPE_DRM_PRIME_3
+#endif
                                     | VA_SURFACE_ATTRIB_MEM_TYPE_DRM_PRIME_2;
 
         // create surfaces for each supported attribute