diff --git a/layouts/shortcodes/p5_sketch.html b/layouts/shortcodes/p5_sketch.html
index eeee38f..c34b2df 100644
--- a/layouts/shortcodes/p5_sketch.html
+++ b/layouts/shortcodes/p5_sketch.html
@@ -1,2 +1,2 @@
 {{ $id := .Get "id" }}
-<div class="sketch-container" id="{{ $id }}"></div>
\ No newline at end of file
+<figure class="p5-sketch" id="{{ $id }}"></figure>
\ No newline at end of file