diff --git a/fp8_optimization.py b/fp8_optimization.py
index 3f7aa9c..1e7c42b 100644
--- a/fp8_optimization.py
+++ b/fp8_optimization.py
@@ -2,7 +2,6 @@
 
 import torch
 import torch.nn as nn
-from types import MethodType
 
 def fp8_linear_forward(cls, original_dtype, input):
     weight_dtype = cls.weight.dtype