2 files changed, 2 insertions, 2 deletions
diff --git a/lib/pb-protocol/pb-protocol.c b/lib/pb-protocol/pb-protocol.c
index ffa2c1b..4981e95 100644
--- a/lib/pb-protocol/pb-protocol.c
+++ b/lib/pb-protocol/pb-protocol.c
@@ -201,7 +201,7 @@ int pb_protocol_write_message(int fd, struct pb_protocol_message *message)
 }
 
 struct pb_protocol_message *pb_protocol_create_message(void *ctx,
-		int action, int payload_len)
+		enum pb_protocol_action action, int payload_len)
 {
 	struct pb_protocol_message *message;
 
diff --git a/lib/pb-protocol/pb-protocol.h b/lib/pb-protocol/pb-protocol.h
index ce9c96b..399c824 100644
--- a/lib/pb-protocol/pb-protocol.h
+++ b/lib/pb-protocol/pb-protocol.h
@@ -52,7 +52,7 @@ int pb_protocol_serialise_device(struct device *dev, char *buf, int buf_len);
 int pb_protocol_write_message(int fd, struct pb_protocol_message *message);
 
 struct pb_protocol_message *pb_protocol_create_message(void *ctx,
-		int action, int payload_len);
+		enum pb_protocol_action action, int payload_len);
 
 struct pb_protocol_message *pb_protocol_read_message(void *ctx, int fd);