fix a bug of gen4go