diff --git a/src/pty.c b/src/pty.c
index 3f52021..5ca2e28 100644
--- a/src/pty.c
+++ b/src/pty.c
@@ -54,8 +54,8 @@ struct kmscon_pty {
 	void *closed_data;
 };
 
-int kmscon_pty_new(struct kmscon_pty **out,
-				kmscon_pty_output_cb output_cb, void *data)
+int kmscon_pty_new(struct kmscon_pty **out, kmscon_pty_output_cb output_cb,
+								void *data)
 {
 	struct kmscon_pty *pty;
 
@@ -71,9 +71,9 @@ int kmscon_pty_new(struct kmscon_pty **out,
 	memset(pty, 0, sizeof(*pty));
 	pty->fd = -1;
 	pty->ref = 1;
-
 	pty->output_cb = output_cb;
 	pty->output_data = data;
+
 	*out = pty;
 	return 0;
 }
diff --git a/src/pty.h b/src/pty.h
index 6cb0877..cc66362 100644
--- a/src/pty.h
+++ b/src/pty.h
@@ -42,6 +42,7 @@
 #ifndef KMSCON_PTY_H
 #define KMSCON_PTY_H
 
+#include <stdlib.h>
 #include "eloop.h"
 
 struct kmscon_pty;
@@ -50,8 +51,8 @@ typedef void (*kmscon_pty_output_cb)
 		(struct kmscon_pty *pty, char *u8, size_t len, void *data);
 typedef void (*kmscon_pty_closed_cb) (struct kmscon_pty *pty, void *data);
 
-int kmscon_pty_new(struct kmscon_pty **out,
-				kmscon_pty_output_cb output_cb, void *data);
+int kmscon_pty_new(struct kmscon_pty **out, kmscon_pty_input_cb input_cb,
+								void *data);
 void kmscon_pty_ref(struct kmscon_pty *pty);
 void kmscon_pty_unref(struct kmscon_pty *pty);