diff --git a/test/crash.erl b/test/crash.erl
index aed4f1f..bb871c9 100644
--- a/test/crash.erl
+++ b/test/crash.erl
@@ -10,8 +10,8 @@
 -export([start/0]).
 
 -record(state, {
-        host,
-        port
+        host :: term(),
+        port :: term()
     }).
 
 start() ->
diff --git a/test/pr_nested_record_test.erl b/test/pr_nested_record_test.erl
index 7a6e284..9e2fbf5 100644
--- a/test/pr_nested_record_test.erl
+++ b/test/pr_nested_record_test.erl
@@ -2,13 +2,11 @@
 
 -compile([{parse_transform, lager_transform}]).
 
--record(a, {field1, field2}).
--record(b, {field1, field2}).
+-record(a, {field1 :: term(), field2 :: term()}).
+-record(b, {field1 :: term() , field2 :: term()}).
 
 
--ifdef(TEST).
 -include_lib("eunit/include/eunit.hrl").
--endif.
 
 nested_record_test() ->
     A = #a{field1 = x, field2 = y},