Fix minor error

during rewrite