diff --git a/src/StippleMakie.jl b/src/StippleMakie.jl
index d547bd3..2ca7d78 100644
--- a/src/StippleMakie.jl
+++ b/src/StippleMakie.jl
@@ -11,7 +11,7 @@ WS_PROXIES = isdefined(Genie.Router, :WS_PROXIES) ? Genie.Router.WS_PROXIES : Di
 
 export MakieFigure, init_makiefigures, makie_figure, makie_dom, configure_makie_server!, WGLMakie, Makie
 
-@kwdef mutable struct MakieFigure
+Base.@kwdef mutable struct MakieFigure
     fig::Figure = Figure()
     session::Union{Nothing, Bonito.Session} = nothing
     id = -1
diff --git a/test/runtests.jl b/test/runtests.jl
index d84c756..697a581 100644
--- a/test/runtests.jl
+++ b/test/runtests.jl
@@ -39,8 +39,12 @@ UI::ParsedHTMLString = column(style = "height: 80vh; width: 98vw", [
 ui() = UI
 
 model = @init MakieDemo    
-html!(ui, layout = Stipple.ReactiveTools.DEFAULT_LAYOUT(head_content = [makie_dom(model)]), model = model, context = @__MODULE__)
+response = html!(ui, layout = Stipple.ReactiveTools.DEFAULT_LAYOUT(head_content = [makie_dom(model)]), model = model, context = @__MODULE__)
 
 model.isready[] = true
 
+@test model isa MakieDemo
+@test response isa Genie.Requests.HTTP.Response
+@test makie_dom(model) isa ParsedHTMLString
+
 nothing
\ No newline at end of file