fix preprocess script

Signed-off-by: Wanpeng Song <swpustc@mail.ustc.edu.cn>