Fix the build.

Not 100% sure if this is the right thing, but this fixes the build for me.