Fix a bug for AOJ