diff --git a/src/enso.nim b/src/enso.nim
index a931016..5d28e00 100644
--- a/src/enso.nim
+++ b/src/enso.nim
@@ -2,8 +2,9 @@ import std/sugar
 
 # Declarations
 
-type FileDesc* = enum STDIN, STDOUT, STDERR
-type IO*[T]    = proc(): T
+type
+  FileDesc* = enum STDIN, STDOUT, STDERR
+  IO*[T]    = proc(): T
 
 func flatmap*[T, U](io: IO[T], fn: T -> IO[U]): IO[U]
 func lift*(fn: proc(): void): IO[void]
diff --git a/tests/test.nim b/tests/test.nim
index 6d79a19..98173ed 100644
--- a/tests/test.nim
+++ b/tests/test.nim
@@ -1,5 +1,5 @@
 import enso
-import std/strformat
+import std/[strformat, sugar]
 
 proc unsafe_void(str: string) =
   echo str
@@ -14,8 +14,8 @@ func main() =
   run strput("Hello from Enso!")
 
   let
-    out_void: IO[void] = lift do(): unsafe_void("void")
-    out_int:  IO[int]  = lift proc(): int = unsafe_int("int")
+    out_void: IO[void] = lift () => unsafe_void("void")
+    out_int:  IO[int]  = lift () => unsafe_int("int")
 
   out_void()
   run strput(fmt"{out_int()}")