diff --git a/loxi_ir/unified.py b/loxi_ir/unified.py
index d481730..7bcdd83 100644
--- a/loxi_ir/unified.py
+++ b/loxi_ir/unified.py
@@ -153,3 +153,4 @@
     unified = ir.OFProtocol(version=None, classes = tuple(unified_classes.values()), enums=unified_enums)
     for e in chain(unified.classes, unified.enums):
         e.protocol = unified
+    return unified