diff --git a/vmoe/nn/vit_moe.py b/vmoe/nn/vit_moe.py
index d9f0277..7319db4 100644
--- a/vmoe/nn/vit_moe.py
+++ b/vmoe/nn/vit_moe.py
@@ -34,7 +34,6 @@
 Shape = Iterable[int]
 
 
-
 # Slight modification of the VisionTransformer's MlpBlock API.
 class MlpBlock(models_vit.MlpBlock):
   dtype: Optional[DType] = None