Fix #152 in aibenchmark.