Fixed some compilation issues

Signed-off-by: Hin C Lam <hincheuk@us.ibm.com>