Fix army calculation