diff --git a/playground/src/Main.hs b/playground/src/Main.hs
index bcd55ca..bdf5a6d 100644
--- a/playground/src/Main.hs
+++ b/playground/src/Main.hs
@@ -364,6 +364,7 @@ withHaskellFile m action = withSystemTempFile "playground.hs" $ \target h -> do
     hClose h
     T.writeFile target
       "{-# LANGUAGE OverloadedStrings #-}\n\
+      \{-# LANGUAGE FlexibleContexts #-}\n\
       \module Animation where\n\
       \import Reanimate\n\
       \import Reanimate.Builtin.Documentation\n\
@@ -378,12 +379,14 @@ withHaskellFile m action = withSystemTempFile "playground.hs" $ \target h -> do
       \import Reanimate.Scene\n\
       \import qualified Graphics.SvgTree as SVG\n\
       \import Control.Lens\n\
+      \import Control.Monad\n\
       \import qualified Data.Text as T\n\
       \import Linear.V2\n\
       \import Linear.Metric\n\
       \import Linear.Vector\n\
       \import Text.Printf\n\
       \import Codec.Picture.Types\n\
+      \import Control.Monad.State\n\
       \-- Used for testing:\n\
       \-- import System.IO.Unsafe\n\
       \-- import Control.Concurrent\n\