diff --git a/src/connection.rs b/src/connection.rs
index 2b279a4..b2eda1a 100644
--- a/src/connection.rs
+++ b/src/connection.rs
@@ -2,7 +2,7 @@
 
 use crate::{message::RawGenlMessage, GenetlinkHandle};
 use futures::channel::mpsc::UnboundedReceiver;
-use netlink_packet_core::NetlinkMessage;
+use netlink_packet_core::{NetlinkEvent, NetlinkMessage};
 use netlink_proto::{
     self,
     sys::{protocols::NETLINK_GENERIC, AsyncSocket, SocketAddr},
@@ -29,7 +29,9 @@ use std::io;
 pub fn new_connection() -> io::Result<(
     Connection<RawGenlMessage>,
     GenetlinkHandle,
-    UnboundedReceiver<(NetlinkMessage<RawGenlMessage>, SocketAddr)>,
+    UnboundedReceiver<
+        NetlinkEvent<(NetlinkMessage<RawGenlMessage>, SocketAddr)>,
+    >,
 )> {
     new_connection_with_socket()
 }
@@ -40,7 +42,9 @@ pub fn new_connection() -> io::Result<(
 pub fn new_connection_with_socket<S>() -> io::Result<(
     Connection<RawGenlMessage, S>,
     GenetlinkHandle,
-    UnboundedReceiver<(NetlinkMessage<RawGenlMessage>, SocketAddr)>,
+    UnboundedReceiver<
+        NetlinkEvent<(NetlinkMessage<RawGenlMessage>, SocketAddr)>,
+    >,
 )>
 where
     S: AsyncSocket,