diff --git a/java_gen/pre-written/src/main/java/org/projectfloodlight/openflow/types/VlanVid.java b/java_gen/pre-written/src/main/java/org/projectfloodlight/openflow/types/VlanVid.java
index b63e4dc..8337eb6 100644
--- a/java_gen/pre-written/src/main/java/org/projectfloodlight/openflow/types/VlanVid.java
+++ b/java_gen/pre-written/src/main/java/org/projectfloodlight/openflow/types/VlanVid.java
@@ -30,10 +30,10 @@
         this.vid = vid;
     }
 
-    public static VlanVid ofVlan(short vid) {
+    public static VlanVid ofVlan(int vid) {
         if ((vid & VALIDATION_MASK) != vid)
             throw new IllegalArgumentException(String.format("Illegal VLAN value: %x", vid));
-        return new VlanVid(vid);
+        return new VlanVid((short) vid);
     }
 
     /** @return the actual VLAN tag this vid identifies */