diff --git a/src/LaMEM_ModelGeneration/LaMEM_ModelGeneration.jl b/src/LaMEM_ModelGeneration/LaMEM_ModelGeneration.jl
index 33a3e98..a95124b 100644
--- a/src/LaMEM_ModelGeneration/LaMEM_ModelGeneration.jl
+++ b/src/LaMEM_ModelGeneration/LaMEM_ModelGeneration.jl
@@ -28,9 +28,10 @@ This returns a string with the documentation for a parameter `field` that is wit
 Note that this structure must be a help structure of the current one.
 """
 function get_doc(structure, field::Symbol) 
-    #=
     alldocs       =   Docs.meta(LaMEM_Model);
     var           =   eval(Meta.parse("Docs.@var($structure)"))
+    @show var
+    #=
     fields_local  =   alldocs[var].docs[Union{}].data[:fields]
     str = fields_local[field]
 
diff --git a/src/MakieExt.jl b/src/MakieExt.jl
index 0b685c1..7b05c70 100644
--- a/src/MakieExt.jl
+++ b/src/MakieExt.jl
@@ -61,7 +61,7 @@ function plot_initial_setup(model::LaMEM.Model; x=nothing, y=nothing, z=nothing,
 
     display(fig)
     =#
-    h = Plots.heatmap(x_vec, z_vec, data, xlabel=x_str, ylabel=z_str, title=title_str, aspect_ratio=:equal)
+    h = heatmap(x_vec, z_vec, data, xlabel=x_str, ylabel=z_str, title=title_str, aspect_ratio=:equal)
     display(h)
 
     return h