Fixed compile error
Fixed compile problem