diff --git a/src/kpro_brokers.erl b/src/kpro_brokers.erl
index 7f1b97e..964a6a4 100644
--- a/src/kpro_brokers.erl
+++ b/src/kpro_brokers.erl
@@ -26,7 +26,9 @@
         ]).
 
 -include("kpro_private.hrl").
+-ifdef(TEST).
 -include_lib("eunit/include/eunit.hrl").
+-endif.
 
 -type endpoint() :: kpro:endpoint().
 -type topic() :: kpro:topic().
diff --git a/src/kpro_connection.erl b/src/kpro_connection.erl
index f48faf1..1df45ed 100644
--- a/src/kpro_connection.erl
+++ b/src/kpro_connection.erl
@@ -42,7 +42,9 @@
              ]).
 
 -include("kpro_private.hrl").
+-ifdef(TEST).
 -include_lib("eunit/include/eunit.hrl").
+-endif.
 
 -define(DEFAULT_CONNECT_TIMEOUT, timer:seconds(5)).
 -define(DEFAULT_REQUEST_TIMEOUT, timer:minutes(4)).