diff --git a/Tapeti/Config/IControllerCleanupMiddleware.cs b/Tapeti/Config/IControllerCleanupMiddleware.cs
index 9cbd234..2f16269 100644
--- a/Tapeti/Config/IControllerCleanupMiddleware.cs
+++ b/Tapeti/Config/IControllerCleanupMiddleware.cs
@@ -6,7 +6,7 @@ namespace Tapeti.Config
     /// <summary>
     /// Denotes middleware that runs after controller methods.
     /// </summary>
-    public interface IControllerCleanupMiddleware
+    public interface IControllerCleanupMiddleware : IControllerMiddlewareBase
     {
         /// <summary>
         /// Called after the message handler method, even if exceptions occured.
diff --git a/Tapeti/Config/IControllerMessageMiddleware.cs b/Tapeti/Config/IControllerMessageMiddleware.cs
index a252081..65e777f 100644
--- a/Tapeti/Config/IControllerMessageMiddleware.cs
+++ b/Tapeti/Config/IControllerMessageMiddleware.cs
@@ -6,7 +6,7 @@ namespace Tapeti.Config
     /// <summary>
     /// Denotes middleware that runs for controller methods.
     /// </summary>
-    public interface IControllerMessageMiddleware
+    public interface IControllerMessageMiddleware : IControllerMiddlewareBase
     {
         /// <summary>
         /// Called after the message has passed any filter middleware and the controller has been instantiated,