diff --git a/k-bandits.py b/k-bandits.py
index 1cfd591..f5ccea5 100644
--- a/k-bandits.py
+++ b/k-bandits.py
@@ -78,11 +78,11 @@ def test(_k_arms: int, _episode_len: int, ql_star, _tie_break, _param: Param) ->
 		_reward = _reward_z[j] + ql_star[_a]
 
 		# calc
-		_nu[_a] = _nu[_a] + 1
+		_nu[_a] += 1
 		if _param.alpha > 0:
-			_qu[_a] = _qu[_a] + _param.alpha * (_reward - _qu[_a])
+			_qu[_a] += _param.alpha * (_reward - _qu[_a])
 		else:
-			_qu[_a] = _qu[_a] + (_reward - _qu[_a]) / _nu[_a]
+			_qu[_a] += (_reward - _qu[_a]) / _nu[_a]
 
 		# Reduce tendency to explore with number of steps (or with age for humans)
 		_epsilon = _param.epsilon * (1 - _param.rho)