Fix build.

I forgot to commit these changes. Sorry!