Fix moves when the king is under attack