diff --git a/openflow/ctl/src/test/java/org/onosproject/openflow/OfMessageAdapter.java b/openflow/ctl/src/test/java/org/onosproject/openflow/OfMessageAdapter.java
index e9b38e3..b7446eb 100644
--- a/openflow/ctl/src/test/java/org/onosproject/openflow/OfMessageAdapter.java
+++ b/openflow/ctl/src/test/java/org/onosproject/openflow/OfMessageAdapter.java
@@ -26,13 +26,21 @@
  * Adapter for testing against an OpenFlow message.
  */
 public class OfMessageAdapter implements OFMessage {
-    @Override
-    public OFVersion getVersion() {
-        return null;
+    OFType type;
+
+    private OfMessageAdapter() {}
+
+    public OfMessageAdapter(OFType type) {
+        this.type = type;
     }
 
     @Override
     public OFType getType() {
+        return type;
+    }
+
+    @Override
+    public OFVersion getVersion() {
         return null;
     }