diff --git a/java_gen/pre-written/src/main/java/org/projectfloodlight/openflow/types/IPAddressWithMask.java b/java_gen/pre-written/src/main/java/org/projectfloodlight/openflow/types/IPAddressWithMask.java
index fc4fa7d..11ef103 100644
--- a/java_gen/pre-written/src/main/java/org/projectfloodlight/openflow/types/IPAddressWithMask.java
+++ b/java_gen/pre-written/src/main/java/org/projectfloodlight/openflow/types/IPAddressWithMask.java
@@ -1,6 +1,5 @@
 package org.projectfloodlight.openflow.types;
 
-import org.projectfloodlight.openflow.types.IPAddress.IpVersion;
 
 public abstract class IPAddressWithMask<F extends IPAddress<F>> extends Masked<F> {
 
@@ -8,7 +7,7 @@
         super(value, mask);
     }
 
-    public abstract IpVersion getIpVersion();
+    public abstract IPVersion getIpVersion();
 
     public static IPAddressWithMask<?> of(String ip) {
         if (ip.indexOf('.') != -1)