fix the wrong infogain bug