diff --git a/lib/Dialect/ESI/runtime/cpp/include/esi/Accelerator.h b/lib/Dialect/ESI/runtime/cpp/include/esi/Accelerator.h
index 365b14f85404..e80d2600a7a4 100644
--- a/lib/Dialect/ESI/runtime/cpp/include/esi/Accelerator.h
+++ b/lib/Dialect/ESI/runtime/cpp/include/esi/Accelerator.h
@@ -82,7 +82,7 @@ class AcceleratorConnection {
   Context &getCtxt() const { return ctxt; }
 
   /// Disconnect from the accelerator cleanly.
-  void disconnect();
+  virtual void disconnect();
 
   /// Request the host side channel ports for a particular instance (identified
   /// by the AppID path). For convenience, provide the bundle type.