Fix a build break