diff --git a/transformations/Unnest.hs b/transformations/Unnest.hs
index 89eb6f2..45d4733 100644
--- a/transformations/Unnest.hs
+++ b/transformations/Unnest.hs
@@ -227,6 +227,7 @@ removeNesting = pass "Pull nested definitions to top level"
 
     canPull :: A.SpecType -> Bool
     canPull (A.Proc _ _ _ _) = True
+    canPull (A.Function {}) = True
     canPull (A.RecordType _ _ _) = True
     canPull (A.Protocol _ _) = True
     canPull (A.ProtocolCase _ _) = True