Fixed build issue.

Fixed build issue.