1 files changed, 2 insertions, 1 deletions
diff --git a/pci.c b/pci.c
index 1ee99ce..c6c37c9 100644
--- a/pci.c
+++ b/pci.c
@@ -52,8 +52,9 @@ static void pcilib_print_error(const char *msg, ...) {
 void (*pcilib_error)(const char *msg, ...) = pcilib_print_error;
 void (*pcilib_warning)(const char *msg, ...) = pcilib_print_error;
 
-int pcilib_set_error_handler(void (*err)(const char *msg, ...)) {
+int pcilib_set_error_handler(void (*err)(const char *msg, ...), void (*warn)(const char *msg, ...)) {
     pcilib_error = err;
+    pcilib_warning = warn;
 }
 
 pcilib_t *pcilib_open(const char *device, pcilib_model_t model) {