Fixed a compile bug