Fixed a build issue