Fixed bug in update of A