I agree with Didz here. If the square is formed before the cavalry arrive then its accurate that the cavalry should get the short end of the stick but if the cavalry get into the ranks before a square is formed then the infantry should suffer. One of the things the AI does wrong is that it will stop a cavalry unit extremely close to a formed square and just sit there and be picked off. Reality is that the cavalry wouldnt get that close and just sit there and be allowed to be killed. They would stay back a few hundred yards at least and be ready to exploit a breakthrough. They wouldnt willingly want to charge a square.