diff --git a/vllm/v1/attention/backends/mla/aiter_triton_mla.py b/vllm/v1/attention/backends/mla/aiter_triton_mla.py
index 96bc0480d..6082a5e86 100644
--- a/vllm/v1/attention/backends/mla/aiter_triton_mla.py
+++ b/vllm/v1/attention/backends/mla/aiter_triton_mla.py
@@ -1,13 +1,9 @@
 # SPDX-License-Identifier: Apache-2.0
 # SPDX-FileCopyrightText: Copyright contributors to the vLLM project
-from vllm.v1.attention.backends.mla.common import MLACommonBackend
-from vllm.v1.attention.backends.mla.rocm_aiter_mla import (
-    AiterMLAImpl,
-    AiterMLAMetadataBuilder,
-)
+from vllm.v1.attention.backends.mla.rocm_aiter_mla import AiterMLABackend, AiterMLAImpl
 
 
-class AiterTritonMLABackend(MLACommonBackend):
+class AiterTritonMLABackend(AiterMLABackend):
     @staticmethod
     def get_name() -> str:
         return "AITER_TRITON_MLA"
@@ -16,10 +12,6 @@ class AiterTritonMLABackend(MLACommonBackend):
     def get_impl_cls() -> type["AiterTritonMLAImpl"]:
         return AiterTritonMLAImpl
 
-    @staticmethod
-    def get_builder_cls() -> type["AiterMLAMetadataBuilder"]:
-        return AiterMLAMetadataBuilder
-
 
 class AiterTritonMLAImpl(AiterMLAImpl):
     def __init__(