diff --git a/py_gen/templates/common.py b/py_gen/templates/common.py
index ccf6032..5245a26 100644
--- a/py_gen/templates/common.py
+++ b/py_gen/templates/common.py
@@ -71,6 +71,10 @@
 
 :: if version == 1:
 match = match_v1
+:: elif version == 2:
+match = match_v2
+:: elif version == 3:
+match = match_v3
 :: elif version == 4:
 :: # HACK
 match = match_v3