sync with paper

minor fix, try to be same as in the paper