diff --git a/tinygrad/tensor.py b/tinygrad/tensor.py
index f8e91f2932..7e4e045c2f 100644
--- a/tinygrad/tensor.py
+++ b/tinygrad/tensor.py
@@ -203,8 +203,7 @@ class Tensor:
 
   def mean(self, axis=None):
     out = self.sum(axis=axis)
-    coeff = np.prod(out.shape)/np.prod(self.shape)
-    return out * coeff
+    return out * (np.prod(out.shape)/np.prod(self.shape))
 
   def sqrt(self):
     return self.pow(0.5)
@@ -242,8 +241,7 @@ class Tensor:
     # TODO: this needs a test
     if Tensor.training:
       _mask = np.asarray(np.random.binomial(1, 1.0-p, size=self.shape), dtype=self.dtype)
-      ret = self * Tensor(_mask, requires_grad=False, device=self.device)
-      return ret * (1/(1.0 - p))
+      return self * Tensor(_mask, requires_grad=False, device=self.device) * (1/(1.0 - p))
     else:
       return self