[bug fix] PTSDK-1166

fix some missing code
Merge pull request #152 from zivchang/next

[bug fix] PTSDK-1166