Fix a compile error