Fix a build issue