Fixed compile failure

Fixed compile failure