diff --git a/src/petals/models/llama/block.py b/src/petals/models/llama/block.py
index e8e36621a..a8d433ded 100644
--- a/src/petals/models/llama/block.py
+++ b/src/petals/models/llama/block.py
@@ -4,7 +4,6 @@
 See commit history for authorship.
 """
 import math
-import math
 from typing import Optional, Tuple
 
 import torch
diff --git a/tests/test_optimized_layers.py b/tests/test_optimized_layers.py
index c66e16e83..84cbfffe2 100644
--- a/tests/test_optimized_layers.py
+++ b/tests/test_optimized_layers.py
@@ -4,7 +4,6 @@
 import torch
 from transformers.models.falcon.modeling_falcon import FalconDecoderLayer, FalconModel, build_alibi_tensor
 from transformers.models.llama.modeling_llama import LlamaDecoderLayer, LlamaModel
-from transformers.models.llama.modeling_llama import LlamaDecoderLayer, LlamaModel
 
 from petals.utils.auto_config import AutoDistributedConfig
 from petals.utils.convert_block import QuantType, convert_block