diff --git a/drivers/battery/battc/battc.c b/drivers/battery/battc/battc.c
index a626d8c9171..3a73bc9c56f 100644
--- a/drivers/battery/battc/battc.c
+++ b/drivers/battery/battc/battc.c
@@ -46,10 +46,10 @@ BCLASSAPI
 NTSTATUS
 DDKAPI
 BatteryClassSystemControl(PVOID ClassData,
-                          PWMILIB_CONTEXT WmiLibContext,
+                          PVOID WmiLibContext,
                           PDEVICE_OBJECT DeviceObject,
                           PIRP Irp,
-                          PSYSCTL_IRP_DISPOSITION Disposition)
+                          PVOID Disposition)
 {
   NTSTATUS Status;
 
diff --git a/drivers/battery/battc/battc.h b/drivers/battery/battc/battc.h
index b7c5af49333..0839cea845e 100644
--- a/drivers/battery/battc/battc.h
+++ b/drivers/battery/battc/battc.h
@@ -9,6 +9,7 @@
 #pragma once
 
 #include <ntddk.h>
+#include <wmlib.h>
 #include <initguid.h>
 #include <batclass.h>
 
diff --git a/include/ddk/wmlib.h b/include/ddk/wmlib.h
index e650b57fb36..6d48b776639 100644
--- a/include/ddk/wmlib.h
+++ b/include/ddk/wmlib.h
@@ -12,6 +12,18 @@ typedef struct _WMIGUIDREGINFO {
   ULONG Flags;
 } WMIGUIDREGINFO, *PWMIGUIDREGINFO;
 
+typedef enum _WMIENABLEDISABLECONTROL {
+  WmiEventControl,
+  WmiDataBlockControl
+} WMIENABLEDISABLECONTROL, *PWMIENABLEDISABLECONTROL;
+
+typedef enum _SYSCTL_IRP_DISPOSITION {
+  IrpProcessed,
+  IrpNotCompleted,
+  IrpNotWmi,
+  IrpForward
+} SYSCTL_IRP_DISPOSITION, *PSYSCTL_IRP_DISPOSITION;
+
 typedef
 NTSTATUS
 (NTAPI WMI_QUERY_REGINFO_CALLBACK)(