fix refactoring error
Merge pull request #324 from hurzl/upstr

fix refactoring error