diff --git a/src/rebar3_erllambda_rlx_zip_prv.erl b/src/rebar3_erllambda_rlx_zip_prv.erl
index 13b2bda..02a3b88 100644
--- a/src/rebar3_erllambda_rlx_zip_prv.erl
+++ b/src/rebar3_erllambda_rlx_zip_prv.erl
@@ -47,18 +47,10 @@ format_error(ErrorDetail) ->
 %% Internal Functions
 %%============================================================================
 archive_path(State, Release) ->
-    Dir = archive_dir(State),
+    Dir = rlx_state:base_output_dir(State),
     FileName = archive_name(Release),
     filename:join(Dir, FileName).
 
-archive_dir(State) ->
-    case get({rebar3_erllambda_rlx_zip_prv, archive_dir}) of
-        undefined ->
-            rlx_state:base_output_dir(State);
-        Dir ->
-            Dir
-    end.
-
 archive_name(Release) ->
     Name = atom_to_list(rlx_release:name(Release)),
     Vsn = rlx_release:vsn(Release),
diff --git a/src/rebar3_erllambda_zip.erl b/src/rebar3_erllambda_zip.erl
index c853f52..d992efd 100644
--- a/src/rebar3_erllambda_zip.erl
+++ b/src/rebar3_erllambda_zip.erl
@@ -54,8 +54,6 @@ do( State ) ->
     rebar_api:info( "generating erllambda zip package", [] ),
     State1 = rebar3_erllambda:add_property(
                State, relx, add_providers, rebar3_erllambda_rlx_zip_prv),
-    %% Internal tooling expects artifacts in this specific location, to be fixed
-    set_archive_dir(State1),
     rebar_relx:do(rebar3_erllambda_rlx_prv, "erllambda_zip", ?PROVIDER, State1).
 
 %%%---------------------------------------------------------------------------
@@ -70,6 +68,3 @@ format_error( Error ) ->
 %%============================================================================
 %% Internal Functions
 %%============================================================================
-set_archive_dir(State) ->
-    BaseDir = rebar_dir:base_dir(State),
-    put({rebar3_erllambda_rlx_zip_prv, archive_dir}, BaseDir).