Support EOneHot

1 files changed, 2 insertions, 2 deletions

diff --git a/src/AST.hs b/src/AST.hs
index 328a670..b9b10ad 100644
--- a/src/AST.hs
+++ b/src/AST.hs
@@ -103,7 +103,7 @@ data Expr x env t where
   -- monoidal operations (to be desugared to regular operations after simplification)
   EZero :: STy t -> Expr x env (D2 t)
   EPlus :: STy t -> Expr x env (D2 t) -> Expr x env (D2 t) -> Expr x env (D2 t)
-  EOneHot :: STy t -> SNat i -> Expr x env (AcIdx t i) -> Expr x env (D2 (AcVal t i)) -> Expr x env (D2 t)
+  EOneHot :: STy t -> SNat i -> Expr x env (AcIdx (D2 t) i) -> Expr x env (AcVal (D2 t) i) -> Expr x env (D2 t)
 
   -- partiality
   EError :: STy a -> String -> Expr x env a
@@ -120,7 +120,7 @@ type family Tup env where
 
 mkTup :: f TNil -> (forall a b. f a -> f b -> f (TPair a b))
       -> SList f list -> f (Tup list)
-mkTup nil _     SNil = nil
+mkTup nil _    SNil = nil
 mkTup nil pair (e `SCons` es) = pair (mkTup nil pair es) e
 
 tTup :: SList STy env -> STy (Tup env)