1 files changed, 9 insertions, 1 deletions
diff --git a/tests/sock.c b/tests/sock.c
index 7d1d226b..6a122cdf 100644
--- a/tests/sock.c
+++ b/tests/sock.c
@@ -10,10 +10,15 @@
 #include "convey.h"
 #include "stubs.h"
 #include "nng.h"
+#include "core/nng_impl.h"
 
 #include <string.h>
 
-TestMain("Socket Operations", {
+Main({
+	nni_init();
+
+	Test("Socket Operations", {
+
 	Convey("We are able to open a PAIR socket", {
 		int rv;
 		nng_socket *sock = NULL;
@@ -194,4 +199,7 @@ TestMain("Socket Operations", {
 			nng_close(sock2);
 		})
 	})
+	})
+
+	nni_fini();
 })