diff --git a/ext/TenetGraphMakieExt.jl b/ext/TenetGraphMakieExt.jl
index e8e121d7e..bd5e974a1 100644
--- a/ext/TenetGraphMakieExt.jl
+++ b/ext/TenetGraphMakieExt.jl
@@ -5,6 +5,7 @@ using Makie
 const Graphs = GraphMakie.Graphs
 using Tenet
 using Combinatorics: combinations
+const NetworkLayout = GraphMakie.NetworkLayout
 
 """
     graphplot(tn::TensorNetwork; kwargs...)
@@ -129,6 +130,7 @@ function GraphMakie.graphplot!(ax::Union{Axis,Axis3}, tn::TensorNetwork; labels=
     end
     get!(() -> repeat([:black], Graphs.ne(graph)), kwargs, :elabels_color)
     get!(() -> repeat([17], Graphs.ne(graph)), kwargs, :elabels_textsize)
+    get!(() -> NetworkLayout.Stress(), kwargs, :layout)
 
     # plot graph
     return graphplot!(ax, graph; kwargs...)