diff --git a/test/dockerl_test.erl b/test/dockerl_test.erl
index bf78babdd2a725944ee46ad1a0f5dcb69ab30511..68b65a2e0aafc162e8d360261cb5cca99028e679 100644
--- a/test/dockerl_test.erl
+++ b/test/dockerl_test.erl
@@ -55,7 +55,7 @@ pull_image(Pid) ->
     ?_assertMatch({ok, _}, dockerl:pull_image(Pid, <<"nginx:alpine">>)).
 
 create_container(Pid, Name) ->
-    ?_assertMatch({ok, _}, dockerl:create_container(Pid, <<"nginx:alpine">>, Name, #{'PublishAllPorts' => true, 'Tty' => true})).
+    ?_assertMatch({ok, _}, dockerl:create_container(Pid, <<"nginx:alpine">>, Name, #{'Tty' => true})).
 
 start_container(Pid, Name) ->
     ?_assertMatch(ok, dockerl:start_container(Pid, Name)).
@@ -63,9 +63,8 @@ start_container(Pid, Name) ->
 query_container(Pid, Id) ->
     fun() ->
         timer:sleep(3000),
-        {ok, Address} = dockerl_utils:get_gateway(Pid, Id),
-        {ok, #{<<"80/tcp">> := [{_, Port}]}} = dockerl_utils:get_ports(Pid, Id),
-        spawn(fun() -> send_request(Address, Port) end),
+        {ok, Address} = dockerl_utils:get_ip_address(Pid, Id),
+        spawn(fun() -> send_request(Address, 80) end),
         {ok, Stream} = dockerl:container_logs(Pid, Id),
         ?assertEqual(ok, match_logs(Stream, <<"eunit">>, 1))
     end.