diff --git a/drivers/optical/src/main/resources/optical-drivers.xml b/drivers/optical/src/main/resources/optical-drivers.xml
index 1dd44ff..a7ef017 100644
--- a/drivers/optical/src/main/resources/optical-drivers.xml
+++ b/drivers/optical/src/main/resources/optical-drivers.xml
@@ -60,7 +60,7 @@
                    impl="org.onosproject.net.optical.DefaultOpticalDevice"/>
         <behaviour api="org.onosproject.net.behaviour.PowerConfig"
                    impl="org.onosproject.driver.optical.power.OplinkRoadmPowerConfig"/>
-        <behaviour api="org.onosproject.codec.ExtensionTreatmentCodec"
+        <behaviour api="org.onosproject.net.flow.ExtensionTreatmentCodec"
                    impl="org.onosproject.driver.optical.extensions.OplinkExtensionTreatmentCodec"/>
         <behaviour api="org.onosproject.openflow.controller.ExtensionTreatmentInterpreter"
                    impl="org.onosproject.driver.extensions.OplinkExtensionTreatmentInterpreter"/>