Code Refactor fix build issue.

Signed-off-by: Pengcheng Bian <pbian@actuate.com>
Merge pull request #54 from TerryPCBian/master

Code Refactor fix build issue.