fix problems for PR #113

Closed johnzhang27 closed 3 years ago