Fixed build issue

Fixed build issue