Compile fix.

Forgot one change during merge.