diff --git a/drivers/src/main/java/org/onosproject/driver/extensions/NiciraResubmitTable.java b/drivers/src/main/java/org/onosproject/driver/extensions/NiciraResubmitTable.java
index 3c8dd0c..4743d21 100644
--- a/drivers/src/main/java/org/onosproject/driver/extensions/NiciraResubmitTable.java
+++ b/drivers/src/main/java/org/onosproject/driver/extensions/NiciraResubmitTable.java
@@ -20,7 +20,8 @@
 
 import org.onlab.util.KryoNamespace;
 import org.onosproject.net.PortNumber;
-import org.onosproject.net.flow.instructions.AbstractExtensionTreatment;
+import org.onosproject.net.flow.AbstractExtension;
+import org.onosproject.net.flow.instructions.ExtensionTreatment;
 import org.onosproject.net.flow.instructions.ExtensionTreatmentType;
 import org.onosproject.store.serializers.PortNumberSerializer;
 
@@ -33,7 +34,8 @@
 /**
  * Nicira resubmit-table extension instruction.
  */
-public class NiciraResubmitTable extends AbstractExtensionTreatment {
+public class NiciraResubmitTable extends AbstractExtension implements
+        ExtensionTreatment {
 
     //the list of the in port number(PortNumber) and the table(short)
     private List<Object> inPortAndTable = new ArrayList<Object>();
@@ -109,4 +111,4 @@
         return MoreObjects.toStringHelper(getClass())
                 .add("inPortAndTable", inPortAndTable).toString();
     }
-}
\ No newline at end of file
+}