Fixed compile issue
Fixed compile problem