Fix a compile issue