diff --git a/driver/build.rs b/driver/build.rs
index cd5eaa0..5933d9d 100644
--- a/driver/build.rs
+++ b/driver/build.rs
@@ -1,4 +1,6 @@
 fn main() -> Result<(), wdk_build::ConfigError> {
-    wdk_build::Config::from_env_auto()?.configure_binary_build();
+    let mut config = wdk_build::Config::from_env_auto()?;
+    config.driver_config = wdk_build::DriverConfig::WDM();
+    config.configure_binary_build();
     Ok(())
-}
+}
\ No newline at end of file