diff --git a/src/qemu/qemu_command.c b/src/qemu/qemu_command.c
index 2ceff15512..be93182092 100644
--- a/src/qemu/qemu_command.c
+++ b/src/qemu/qemu_command.c
@@ -6804,7 +6804,8 @@ qemuBuildCpuCommandLine(virCommand *cmd,
 
 static bool
 qemuAppendKeyWrapMachineParm(virBuffer *buf, virQEMUCaps *qemuCaps,
-                             int flag, const char *pname, int pstate)
+                             virQEMUCapsFlags flag, const char *pname,
+                             virTristateSwitch pstate)
 {
     if (pstate != VIR_TRISTATE_SWITCH_ABSENT) {
         if (!virQEMUCapsGet(qemuCaps, flag)) {