diff --git a/ktransformers/operators/linear.py b/ktransformers/operators/linear.py
index 96d3578..103fc1a 100644
--- a/ktransformers/operators/linear.py
+++ b/ktransformers/operators/linear.py
@@ -347,7 +347,6 @@ class KLinearMarlin(KLinearBase):
             orig_shape[-1] = self.out_features
         if self.has_bias:
             x = x + self.bias
-        orig_shape[-1] = self.n
         return x.reshape(orig_shape).to(orig_dtype)
 
     def unload(self):