Fixed a compile issue