get lensy and classy

1 files changed, 5 insertions, 5 deletions

diff --git a/app/Env.hs b/app/Env.hs
index 82bf9d0..592d608 100644
--- a/app/Env.hs
+++ b/app/Env.hs
@@ -1,14 +1,14 @@
 module Env where
 
-import Code (Interp(..), InterpFn, PrlgEnv)
-import Control.Monad.Trans.State.Lazy (gets, modify)
+import Code (InterpFn, PrlgEnv)
+import CodeLens
 import qualified IR
+import Lens.Family2.State
 
 withStrTable :: (IR.StrTable -> (IR.StrTable, a)) -> Env.PrlgEnv a
 withStrTable f = do
-  st <- gets strtable
-  let (st', x) = f st
-  modify (\s -> s {strtable = st'})
+  (st', x) <- strtable `uses` f
+  strtable .= st'
   return x
 
 findStruct :: String -> Int -> Env.PrlgEnv IR.Id