diff --git a/src/KafkaFlow.IntegrationTests/Core/Exceptions/ErrorExecutingMiddlewareException.cs b/src/KafkaFlow.IntegrationTests/Core/Exceptions/ErrorExecutingMiddlewareException.cs
index e649d84e3..a582da1e5 100644
--- a/src/KafkaFlow.IntegrationTests/Core/Exceptions/ErrorExecutingMiddlewareException.cs
+++ b/src/KafkaFlow.IntegrationTests/Core/Exceptions/ErrorExecutingMiddlewareException.cs
@@ -2,7 +2,7 @@
 {
     using System;
 
-    internal class ErrorExecutingMiddlewareException : Exception
+    public class ErrorExecutingMiddlewareException : Exception
     {
         public ErrorExecutingMiddlewareException(string middlewareName)
             : base($"Exception thrown executing {middlewareName}")
diff --git a/src/KafkaFlow.IntegrationTests/Core/Exceptions/PartitionAssignmentException.cs b/src/KafkaFlow.IntegrationTests/Core/Exceptions/PartitionAssignmentException.cs
index 6c67c963e..6f11c60da 100644
--- a/src/KafkaFlow.IntegrationTests/Core/Exceptions/PartitionAssignmentException.cs
+++ b/src/KafkaFlow.IntegrationTests/Core/Exceptions/PartitionAssignmentException.cs
@@ -2,7 +2,7 @@
 {
     using System;
 
-    internal class PartitionAssignmentException : Exception
+    public class PartitionAssignmentException : Exception
     {
         private const string ExceptionMessage = "Partition assignment hasn't occurred yet.";
 
diff --git a/src/KafkaFlow.IntegrationTests/GlobalEventsTest.cs b/src/KafkaFlow.IntegrationTests/GlobalEventsTest.cs
index c7a78d7fe..01c0c5e43 100644
--- a/src/KafkaFlow.IntegrationTests/GlobalEventsTest.cs
+++ b/src/KafkaFlow.IntegrationTests/GlobalEventsTest.cs
@@ -177,7 +177,7 @@ private void ConfigureConsumer<T>(IConsumerConfigurationBuilder consumerConfigur
                     middlewares => middlewares
                         .AddSerializer<ProtobufNetSerializer>()
                         .Add<T>())
-                .WithPartitionsAssignedHandler((_, partitions) =>
+                .WithPartitionsAssignedHandler((_, _) =>
                 {
                     this.isPartitionAssigned = true;
                 });
diff --git a/src/KafkaFlow.IntegrationTests/OpenTelemetryTests.cs b/src/KafkaFlow.IntegrationTests/OpenTelemetryTests.cs
index 2cd2f0fd0..89d60fa3e 100644
--- a/src/KafkaFlow.IntegrationTests/OpenTelemetryTests.cs
+++ b/src/KafkaFlow.IntegrationTests/OpenTelemetryTests.cs
@@ -161,7 +161,7 @@ private async Task<IServiceProvider> GetServiceProvider()
                                                 middlewares => middlewares
                                                     .AddCompressor<GzipMessageCompressor>()
                                                     .Add<GzipMiddleware>())
-                                            .WithPartitionsAssignedHandler((_, partitions) =>
+                                            .WithPartitionsAssignedHandler((_, _) =>
                                             {
                                                 this.isPartitionAssigned = true;
                                             })))