diff --git a/plugins/node/instrumentation-kafkajs/src/types.ts b/plugins/node/instrumentation-kafkajs/src/types.ts
index f0be80bc3c..8cd9a64e03 100644
--- a/plugins/node/instrumentation-kafkajs/src/types.ts
+++ b/plugins/node/instrumentation-kafkajs/src/types.ts
@@ -24,17 +24,17 @@ export interface KafkajsMessage {
   timestamp?: string;
 }
 
-export interface MessageInfo {
+export interface MessageInfo<T = KafkajsMessage> {
   topic: string;
-  message: KafkajsMessage;
+  message: T;
 }
 
-export interface KafkaProducerCustomAttributeFunction {
-  (span: Span, info: MessageInfo): void;
+export interface KafkaProducerCustomAttributeFunction<T = KafkajsMessage> {
+  (span: Span, info: MessageInfo<T>): void;
 }
 
-export interface KafkaConsumerCustomAttributeFunction {
-  (span: Span, info: MessageInfo): void;
+export interface KafkaConsumerCustomAttributeFunction<T = KafkajsMessage> {
+  (span: Span, info: MessageInfo<T>): void;
 }
 
 export interface KafkaJsInstrumentationConfig extends InstrumentationConfig {