diff --git a/soft_keymaster_device.cpp b/soft_keymaster_device.cpp
index eddbc5d..c7605f4 100644
--- a/soft_keymaster_device.cpp
+++ b/soft_keymaster_device.cpp
@@ -269,6 +269,7 @@
     // keymaster2 APIs
     memset(&km2_device_, 0, sizeof(km2_device_));
 
+    km2_device_.flags = flags;
     km2_device_.context = this;
 
     km2_device_.common.tag = HARDWARE_DEVICE_TAG;