Fix compiling

Oops.