Fixed compile error

Fixed compile error