diff --git a/zeta/nn/attention/mgqa.py b/zeta/nn/attention/mgqa.py
index ddaac8fb..cf609191 100644
--- a/zeta/nn/attention/mgqa.py
+++ b/zeta/nn/attention/mgqa.py
@@ -1,4 +1,3 @@
-from dataclasses import dataclass
 from typing import Tuple
 
 import torch