Small update optimization
thanks to @ray176 :)